基于对抗擦除和背景抑制的弱监督目标定位方法

作者：高英; 吴朝捷; 谢欣言; 李明阳; 蔡文天

来源：2022-09-28, 中国, CN202211192618.3.

本发明提供的基于对抗擦除和背景抑制的弱监督目标定位方法,包括以下步骤：使用卷积神经网络作为主干网络,将其末端的池化层和全连接层更换为全局平均池化层和全连接层,将训练集图像输入结构更换后的所述卷积神经网络,进行第一次前向传播,输出特征图F；计算得出类别激活图CAM,基于类别激活图CAM计算得到二维图CAM～(norm)和擦除图M～e；进行擦除操作；得到两次前向传播各自的重要性图；训练卷积神经网络；将待检测图像输入训练后的卷积神经网络中,得到目标定位结果。本发明能够引导神经网络发掘容易在分类任务中被抑制表达的一般性目标区域,并且减少作为非目标区域的背景的激活,最终生成精准覆盖完整目标区域的定位图,有效地实现弱监督目标定位任务。

• 单位
  华南理工大学