基于弱监督和半监督学习的红外舰船分割方法

• 作者: 尹旭 刘兆英 张婷 李玉鑑
• 来源期刊: 山东大学学报（工学版）
• 学科: 工学
• 关键词: 红外舰船 语义分割 弱监督学习 半监督学习 卷积神经网络
• 年，卷（期）: 2022,（2）
• 所属期刊栏目: 机器学习与数据挖掘|Machine Learning & Data Mining
• 页码范围: 99-106
• 页数: 8页
• 分类号: TP3
• 语种: 中文
• DOI: 10.6040/j.issn.1672-3961.0.2021.329

摘要

为降低获取像素级标签的成本,提出一种基于弱监督和半监督学习的红外舰船分割方法,在残差网络(residual network,ResNet)的基础上,设计一个自适应定位模块,并使用相似损失、前景损失和背景损失训练自适应定位模块,生成舰船定位图;利用少量像素级标签数据和大量定位图数据交替训练显著性网络生成显著图;用条件随机场优化显著图,并结合图像级标签生成伪标签图像,使用伪标签图像训练分割网络,得到红外舰船的分割结果.在红外舰船数据集上的平均交并比为71.18％,与当前其他先进方法进行对比,平均交并比提高了9.47％,试验结果表明自适应定位模块能够有效定位红外舰船,交替训练方法可以使红外舰船的边缘更准确.