基于注意力机制特征重建网络的舰船目标检测

• 作者: 牛戈 陈小前 季明江 郭鹏宇 刘勇 冉德超
• 来源期刊: 上海航天（中英文）
• 关键词: 遥感图像 舰船检测 小目标 深度学习 注意力机制
• 年，卷（期）: 2021,（4）
• 所属期刊栏目: 图像处理技术|Image Processing Technology
• 页码范围: 128-136
• 页数: 9页
• 分类号: TP751.1
• 语种: 中文
• DOI: 10.19328/j.cnki.2096⁃8655.2021.04.017

摘要

深度学习为遥感领域诸多应用提供了重要的技术支撑,光学遥感图像的舰船目标检测对国防侦察和预警具有重要意义.真实场景中的舰船往往呈不同方向任意排列,且小目标的占比大,经典的深度学习目标检测算法在这种复杂条件下精度低、易漏检.为此,本文设计了基于注意力机制特征重建网络的舰船目标检测算法.首先,通过引入注意力机制对多尺度特征融合网络模型进行训练,以高召回率产生水平锚框;然后,旋转锚框以缓解密集排列目标引起的噪声问题,并利用特征重建模块来缓解特征不对齐的问题,实现模型精炼.在HRSC2016和DOTA数据集上的测试结果表明:舰船目标检测平均精度分别达到90.20和87.52,相比经典的深度学习目标检测算法得到了有效提升,并在模拟星载嵌入式智能图像处理平台上验证了算法在轨应用的可行性.