一种基于特征重用金字塔的舰船检测算法

作者：卢鑫凯; 王肖霞^*; 杨风暴; 刘哲

来源：电子测量技术, 2022, 45(16): 109-115.

DOI：10.19651/j.cnki.emt.2209295

• 舰船目标
• SAR图像
• K-means
• 注意力机制
• 特征重用
• YOLOV4-tiny

针对现有算法在SAR图像舰船目标检测场景中难以提取模糊目标特征的问题，提出一种基于特征重用金字塔的舰船目标检测算法。所提算法以YOLOV4-tiny为主体，首先将线性因子引入到K-means算法中整合初始锚框，加强网络对多尺度目标的适应性；其次在主干CSPDarknet53-tiny中添加注意力机制来抑制干扰信息，减弱复杂背景的影响；最后利用特征重用机制强化特征金字塔，提升网络对模糊目标特征的提取能力。实验结果表明，相较于YOLOV4-tiny网络，改进后的算法在SSDD数据集上的平均检测精度提升11.79%,证明了改进后算法在舰船检测中的有效性。

• 单位
  中北大学