3 个月前

使用强数据增强的半监督语义分割简单基线

使用强数据增强的半监督语义分割简单基线

摘要

近年来,语义分割领域取得了显著进展。然而,监督式语义分割的成功通常依赖于大量标注数据,而获取这些数据既耗时又成本高昂。受图像分类任务中半监督学习方法成功的启发,本文提出了一种简单但高效的半监督语义分割框架。我们证明了“细节决定成败”:一系列简洁的设计与训练技巧可协同显著提升半监督语义分割的性能。以往工作[3, 27]未能高效利用强数据增强进行伪标签学习,原因是强增强带来的分布显著变化会破坏批归一化(batch normalization)的统计特性。为此,我们提出一种新型批归一化方法——分布特定批归一化(Distribution-Specific Batch Normalization, DSBN),有效解决了该问题,并验证了强数据增强在语义分割中的关键作用。此外,我们设计了一种自校正损失函数(self-correction loss),具有良好的抗噪声能力。通过一系列消融实验,我们充分验证了各组件的有效性。所提方法在Cityscapes和Pascal VOC数据集的半监督设置下均取得了当前最优(state-of-the-art)性能。

代码仓库

基准测试

基准方法指标
semi-supervised-semantic-segmentation-on-1SimpleBaseline(DeepLabv3+ with ImageNet pretrained Xception65, single scale inference)
Validation mIoU: 77.8%
semi-supervised-semantic-segmentation-on-2SimpleBaseline(DeeplabV3+ with ImageNet pretrained Xception65, sinle scale inference)
Validation mIoU: 74.1%
semi-supervised-semantic-segmentation-on-8SimpleBaseline(DeepLabv3+ with ImageNet pretrained Xception65, single scale inference)
Validation mIoU: 78.7%

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
使用强数据增强的半监督语义分割简单基线 | 论文 | HyperAI超神经