基于注意力机制图像块度量学习的弱监督语义分割方法

作者：陈百基; 陈可可; 谢东欣

来源：2021-12-22, 中国, ZL202111578620.X.

本发明公开了一种基于注意力机制图像块度量学习的弱监督语义分割方法,包括：1)基于深度卷积网络的分类网络得到类别激活图像；2)利用类别激活图像得到包含物体定位信息的种子区域与候选区域；3)基于注意力机制分别得到包含前景和背景的图像块提案；4)提取正、负图像块对作为正、负样本对；5)为分类网络增加度量学习任务；6)训练种子区域扩张网络,配合随机游走算法扩大种子区域；7)训练基于深度学习的全监督语义分割网络,得到最终的语义分割掩码。本发明方法引入度量学习产生用于训练的数据,可以兼容各种全监督语义分割网络,在弱监督语义分割中提高生成的种子区域对前景目标物体的覆盖精度,进而提高生成的语义分割掩码的整体精度。

• 单位
  华南理工大学