4 个月前

对抗学习在半监督语义分割中的应用

对抗学习在半监督语义分割中的应用

摘要

我们提出了一种使用对抗网络进行半监督语义分割的方法。大多数现有的判别器都是在图像级别上训练,以区分输入图像是真实还是伪造的,而我们设计了一个全卷积判别器,能够在考虑空间分辨率的情况下区分预测的概率图与真实分割分布。我们展示了所提出的判别器可以通过将对抗损失与模型的标准交叉熵损失结合来提高语义分割的准确性。此外,全卷积判别器通过发现未标记图像预测结果中的可信区域,实现了半监督学习,从而提供了额外的监督信号。与现有利用弱标记图像的方法不同,我们的方法利用未标记图像来增强分割模型。在PASCAL VOC 2012和Cityscapes数据集上的实验结果证明了所提算法的有效性。

代码仓库

KookHoiKim/AdaptSegNet
pytorch
GitHub 中提及
CuberrChen/AdvSemiSeg-Paddle
paddle
GitHub 中提及
lym29/DASeg
pytorch
GitHub 中提及
xiaowillow/AdaptSegNet
pytorch
GitHub 中提及
wasidennis/AdaptSegNet
pytorch
GitHub 中提及
xiaowillow/AdaptSegNet1
pytorch
GitHub 中提及
jizongFox/ReproduceAdaptSegNet
pytorch
GitHub 中提及
VilledeMontreal/urban-segmentation
pytorch
GitHub 中提及
ZHKKKe/PixelSSL
pytorch
GitHub 中提及
Sshanu/AdaptSegNet
pytorch
GitHub 中提及
hfslyc/AdvSemiSeg
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
semi-supervised-semantic-segmentation-on-1Adversarial (DeepLab v2 ImageNet pre-trained)
Validation mIoU: 60.5%
semi-supervised-semantic-segmentation-on-2Adversarial (DeepLab v2 ImageNet pre-trained)
Validation mIoU: 57.1%
semi-supervised-semantic-segmentation-on-4Adversarial
Validation mIoU: 64.3%
semi-supervised-semantic-segmentation-on-5Adversarial (DeepLab v2 ImageNet pre-trained)
Validation mIoU: 59.1%
semi-supervised-semantic-segmentation-on-6Adversarial (DeepLab v2 ImageNet pre-trained)
Validation mIoU: 49.2%
semi-supervised-semantic-segmentation-on-8Adversarial (DeepLab v2 ImageNet pre-trained)
Validation mIoU: 65.70%

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
对抗学习在半监督语义分割中的应用 | 论文 | HyperAI超神经