面向复杂海况和轻量化特性的SAP-YOLOv8船舶目标检测算法

SAP-YOLOv8 Ship Target Detection Algorithm for Complex Sea Conditions and Lightweight Features

  • 摘要: 【目的】YOLOv8算法凭借其高效处理能力和优异检测效果在目标检测领域广受认可,但在船舶目标检测中仍存在抗干扰能力不足、多尺度特征提取效果不佳及参数量偏大等问题,为此提出一种特征增强的轻量级SAP-YOLOv8目标检测算法。【方法】在普通卷积中融入空间深度转换卷积与膨胀卷积,构建SDGD模块,提升对海杂波干扰的抑制能力以及多尺度特征的提取能力;引入RT-DETR中的AIFI模块,取代SPPF模块以增强复杂海况下的上下文建模能力;为优化算法的计算效率,基于部分卷积与坐标注意力机制的原理,构建C3k2_PCCA模块以降低参数量和复杂度,同时提升算法在复杂海况下的轻量化性能和运行效率。【结果】在公开数据集HRSID上进行实验,结果表明SAP-YOLOv8算法与原始算法的参数量几乎相同,但精确率、召回率以及平均精度均值分别提高了1.5%、0.7%、1.6%,且检测效果明显优于其他经典算法。【结论】SAP-YOLOv8算法具有更高的检测精度和运行效率,且能够在复杂海况下表现出更强的鲁棒性和实用价值。

     

    Abstract: Objectives The YOLOv8 algorithm has been widely recognized in the field of object detection due to its efficient processing and superior detection performance. However, in ship target detection problems, it still suffers from insufficient robustness, suboptimal multi-scale feature extraction, and an excessive number of parameters. To overcome these issues, a lightweight SAP-YOLOv8 algorithm with enhanced features representation is proposed. Methods Incorporating spatial depthwise convolution and dilated convolution into standard convolution to construct the SDGD module, thereby enhancing the suppression capability against sea clutter interference and the extraction capability of multi-scale features. Introduce the AIFI module from RT-DETR to replace the SPPF module, thereby enhancing the algorithm's contextual modeling capability in complex sea conditions. To optimize computational efficiency, a C3k2_PCCA module, based on the partial convolution and coordinate attention mechanisms, is constructed to reduce parameters and complexity while improving lightweight performance and runtime efficiency in complex sea conditions. Results Experiments on the public HRSID dataset show that, with nearly unchanged model size, SAP-YOLOv8 improves precision, recall, and mean average precision by 1.5%, 0.7%, and 1.6%, respectively, compared to the original algorithm, and it outperforms other classic algorithms in detection performance. Conclusions The SAP-YOLOv8 algorithm exhibits higher detection accuracy and operational efficiency, while demonstrating stronger robustness and practical value in complex sea conditions.

     

/

返回文章
返回