改进YOLOv5的SAR图像舰船目标检测

作者：谭显东; 彭辉

来源：计算机工程与应用, 2022, 58(04): 247-254.

• 合成孔径雷达(SAR)
• YOLOv5
• 舰船检测
• 坐标注意力机制

近年来针对合成孔径雷达(synthetic aperture radar,SAR)图像中缺乏颜色和纹理细节的舰船检测技术在深度学习领域中得到了广泛研究,利用深度学习技术可以有效避免传统的复杂特征设计,并且检测精度得到极大改善。针对舰船目标检测框具有高长宽比和密集排列问题,提出一种基于改进YOLOv5的目标检测方法。该方法针对舰船目标检测框特点将检测框长宽作为参数进行综合考虑并对损失函数进行曲线优化,并结合坐标注意力机制(coordinate attention,CA),在模型轻量化的同时实现对舰船目标检测的高速与高精度并存。实验结果表明:相比原YOLOv5方法,该方法的检测精度由原来的92.3%提升到96.7%,mAP(mean average precision)指标由原来的92.5%提升到97.2%,明显优于对比方法。通过改进检测框损失函数和特征提取方式,提高对SAR图像中舰船目标的检测效果。

• 出版日期2022
• 单位成都信息工程大学