0%

【2020】ClassMix: Segmentation-Based Data Augmentation for Semi-Supervised Learning

Code: https://github.com/WilhelmT/ClassMix

Motivation

本文贡献

  • 引入了一种新的语义分割增强策略,ClassMix
  • 利用一致性正则化和伪标签进行语义分割
  • 在Cityscapes数据集和Pascal VOC数据集的半监督学习中都具有一定的有效性。

Methods

在无标注数据中采样两张图像A和B,使用分割模型分别进行预测,在A的预测结果中通过argmax生成预测结果,随机选取一半预测标签作为mask,将A图中mask区域合并到B图中。

Loss部分,有标注部分使用真实标签,无标注部分使用生成的伪标签。

Experiments