科研之友
微信
新浪微博
Facebook
分享链接
摘要
针对合成孔雷达（Synthetic Aperture Radar， SAR）图像中小目标舰船和复杂背景下舰船检测精度低的问题，提出一种改进RTMDet（Real-Time Models for object Detection）的SAR舰船检测算法。先优化主干网络结构中的基本构建单元，并引入全局注意力机制SimAM，在不增加额外参数的情况下提高模型对关键特征信息的提取能力；为了在特征融合过程中减少小目标特征信息流失和增加其在浅层特征图的融合占比，构建新的轻量级特征融合模块SPD-RPAFPN；最后在预测模块中将回归损失函数替换为KFIoU，提高模型对小目标舰船的检测能力。在公开数据集RSDD上进行了实验对比，改进后的算法相比RTMDet， 在模型参数量和计算量分别下降4.5%和10.8%的同时，近岸AP值提高了14.6%，总AP值提高2.7%到达90.7%。与目前的主流算法对比，SAR舰船检测精度也明显提升，证明了改进RTMDet算法的有效性。
