特征图强化网络:利用特征图强化船舶检测模型训练的网络结构 |
| |
引用本文: | 张泽琨,谭震彪,余坤,方斌,黄骁,马杰.2023.特征图强化网络:利用特征图强化船舶检测模型训练的网络结构.遥感学报,27(12): 2697-2705DOI:10.11834/jrs.20221656 |
| |
作者姓名: | 张泽琨 谭震彪 余坤 方斌 黄骁 马杰 |
| |
作者单位: | 1.华中科技大学 人工智能及自动化学院 多谱信息技术处理国家重点实验室, 武汉 430072;2.中国舰船研究设计中心, 武汉 430064;3.中铁第四勘察设计院集团有限公司, 武汉 430063 |
| |
基金项目: | 国家自然科学基金(U1913602) |
| |
摘 要: | 随着人工智能技术的进一步发展,深度学习方法在船舶检测领域发挥着重要作用。然而,深度学习算法出现的虚警和漏检,对船舶检测领域技术的应用存在一定的阻碍。虽然经典的深度学习方法能够有效处理单一背景的海面,但是当面对复杂背景之下的数据时,经典模型很容易得出岸上的虚警。并且在常规训练中,模型常常对一些显著特征过于关注,出现特征过拟合现象,当这些显著特征发生改变时极易出现漏检。在模型对输入进行前向传播的过程中,模型中不同网络层会对输入生成对应的映射,也就是特征图。充分利用特征图的语义信息和空间信息是一种有效减少虚警和漏检的方法。与传统模型相比,我们提出的特征图强化网络可以充分利用特征图生成自适应特征图掩码与水陆分割掩码,避免模型的特征过拟合与削弱复杂背景造成的影响,最终达到减少虚警与漏检的目的。在现有公共数据集上与算法模型的对比实验结果表明,本文所提出的方法的性能更为出色,超过了其他SOTA算法。
|
关 键 词: | 遥感成像 人工智能 船舶目标检测 神经网络 特征图 网络强化 旋转目标检测 抑制过拟合 |
收稿时间: | 2021-10-20 |
|
| 点击此处可从《遥感学报》浏览原始摘要信息 |
|
点击此处可从《遥感学报》下载免费的PDF全文 |
|