基于对抗学习和一致性正则的半监督语义分割方法

SEMI-SUPERVISED SEMANTIC SEGMENTATION METHOD BASED ON ADVERSARIAL LEARNING AND CONSISTENT REGULARIZATION

  • 摘要: 为了降低语义分割任务对像素级标签的需求,提出一种基于对抗学习和Mean teachers模型的半监督语义分割方法。该方法训练过程分为两个阶段,第一阶段在分割网络之后连接判别网络,通过对抗学习使分割网络预测结果逐渐接近真实标签;第二阶利用第一阶段的网络参数做指数移动平均得到教师网络,与分割网络做一致性训练,使模型性能进一步提升。使用PASCAL VOC 2012数据集进行实验,结果表明在使用相同数量的标签训练下,该方法的分割图的质量和评价指标mIoU优于现有半监督语义分割方法。

     

    Abstract: In order to reduce the need of pixel level label in semantic segmentation task, this paper proposes a semi-supervised semantic segmentation method based on adversarial learning and Mean teachers model. The training process of this method was divided into two stages. In the first stage, the discriminating network was connected after the segmented network, and the prediction results of the segmentation network were gradually closer to the true label through adversarial learning. In the second stage, the network parameters of the first stage were used to do exponential moving average to get the teacher network to train the consistency with the segmented network, so that the performance of the model was further improved. Experiments on Pascal VOC 2012 data set show that under the same number of label training, the proposed method outperforms the existing semi-supervised semantic segmentation in quality of segmentation graph and evaluating indicator mIoU.

     

/

返回文章
返回