基于改进YOLOv8的SAR图像船舰小目标检测算法

    Ship Small Target Detection Algorithm in SAR Image Based on Improved YOLOv8

    • 摘要: 船舰作为海上战斗的主导力量之一,已成为军事作战的重要检测目标。针对复杂海洋大场景下合成孔径雷达(SAR)图像船舰小尺度目标识别能力低下的问题,提出一种基于改进YOLOv8的SAR图像船舰小目标检测算法。首先,将网络卷积模块Conv替换成空间深度非步进卷积,保留判别特征信息,从而提高小目标检测精度;然后,以协调注意力机制为基础,设计出并行协调注意力机制,加强空间中的相互作用,在背景复杂的SAR图像中对特征信息进行针对性提取并运算;最后,将中心点对角线距离度量和端点距离度量分别融入到SIoU距离损失函数和形状损失函数中,重新定义损失函数DSIoU,以提高网络的收敛速度和检测精度。在高分辨率SAR图像数据集上进行了大量实验,结果表明,相较于YOLOv8算法,改进后算法精度提高了2.9 %,参数量降低了10.6 %,帧率提升了13.3 %,能够满足船舰目标实时监测的需求。

       

      Abstract: As one of the leading forces in maritime battles, ships have become important detection targets in military operations. Aiming at the problems of low recognition abilities of small-scale targets in synthetic aperture radar (SAR) images of ships in large complex ocean scenes, a small target detection algorithm for ships in SAR images based on improved YOLOv8 is proposed. First, the network convolution module Conv is replaced with spatial depth non-stepped convolution to retain discriminative feature information and improve small target detection accuracy; then, based on the coordinated attention mechanism, a parallel coordinated attention mechanism is designed to strengthen the interaction in space, and perform targeted extraction and calculation of feature information in SAR images with complex backgrounds; finally, the diagonal distance measurement of the center point and the endpoint distance measurement are integrated into the distance loss function and shape loss function of SIoU respectively, and the loss function DSIoU is redefined to improve the convergence speed and detection accuracy of the network. A large number of experiments are conducted on high-resolution SAR images dataset. The results show that compared with the YOLOv8 algorithm, the accuracy of the improved algorithm is increased by 2.9 %, the number of parameters is reduced by 10.6 %, and the frame rate is increased by 13.3 %, which can meet the real-time monitoring requirements of ship targets.

       

    /

    返回文章
    返回