基于改进YOLOv5的复杂场景下SAR图像船舶检测方法

A Ship Detection Method for SAR Images in Complex Scene Based on Improved YOLOv5

  • 摘要: 针对复杂场景下合成孔径雷达图像船舶检测中易产生漏检的问题,本文提出了一种基于改进YOLOv5的复杂场景下SAR图像船舶检测算法。该算法首先将由通道注意力和空间注意力共同组成的自适应注意力模块引入YOLOv5的特征提取网络中,通过将特征向量筛选加权后,使重要的目标特征占有更大的网络处理比重,以此增强网络对目标区域的特征学习能力。然后根据SAR图像特性优化了检测模型的损失函数,提升了预测框的置信度,最终降低了复杂场景区域的目标漏检率。实验表明,相比传统YOLOv5算法,本文算法显著提升了召回率。对于复杂场景下的SAR图像船舶目标检测,平均准确率达到了79.8%,相比于传统YOLOv5算法和Faster R-CNN算法分别提高了26.1%和17.3%。

     

    Abstract: The traditional method of synthetic aperture radar image for ship detection is easily affected by complex background, resulting in the number increasing of missed targets. In this paper, the adaptive attention module composed of channel attention and spatial attention is employed for the YOLOv5 network. Most of network is allocated to target feature by using the weighted feature vectors in which the network’s ability of feature learning has been enhanced. Then the loss function of the detection model is optimized where the confidence of prediction boxes has been improved. Also the false negative probability has been reduced for target detecting in the complex background. According to simulations, the recall rate has been improved significantly by the proposed algorithm. The AP (Average Precision) value of ship detection for SAR image in complex scene has reached up to 79.8%, which increased by 26.1% and 17.3% respectively compared with the YOLOv5 algorithm and Faster R-CNN algorithm.

     

/

返回文章
返回