基于改进YOLOv8的红外船舶目标检测算法

Infrared Ship Target Detection Algorithm Based on the Improved YOLOv8

  • 摘要: 针对红外成像中,运动船舶目标因缺乏明确的尺寸、形状和纹理等直观信息,造成在复杂海天背景下检测精度低、小目标误检漏检等情况,本文提出一种改进YOLOv8的红外船舶检测方法。首先设计混合卷积模块,用动态偏移技术调整采样位置来应对目标形状的多样性,通过提供卷积核任意数量的参数来更有效地适应不同尺度的目标特征;其次在SPPF模块后引入SimAM无参数注意力机制,聚焦图像的关键区域并降低计算复杂度;然后优化非极大值抑制,降低大目标检测框对小目标检测框的抑制而导致的漏检情况;最后优化损失函数,采用EIoU替换原损失函数,优化目标定位从而提升检测精度。实验结果表明,改进后的算法在红外船舶数据集上mAP可达到94.67%,与改进前mAP相比提高3.83%。与其他经典算法相比,各评价指标均有不同程度提高,改进后的算法检测效果更好且能够满足红外目标实时检测任务。

     

    Abstract: In infrared imaging, moving ship targets often lack clear intuitive information, such as size, shape, and texture, resulting in low detection accuracy and false detections of small targets in complex sea and sky backgrounds. Therefore, this study proposes an infrared ship detection method by improving YOLOv8. First, a hybrid convolution module was designed, and dynamic offset technology was used to adjust the sampling position to address the diversity of target shapes and better adapt to target characteristics at different scales by providing multiple convolution kernel parameters. Second, the SimAM parameter-free attention mechanism was introduced after SPPF to focus on key areas of the image while reducing computational complexity. The non-maximum suppression algorithm was then optimized to reduce missed detections caused by the suppression of small target detection boxes by large target detection boxes. Finally, the loss function was optimized by replacing the original loss function with EIoU to improve target localization accuracy and overall detection performance. Experimental results show that the improved algorithm achieved a 94.67% mAP on the infrared ship dataset, representing an increase of 3.83% compared with the original mAP. Compared with other classic algorithms, each evaluation index was improved to varying degrees, which demonstrates that the improved algorithm achieves better detection performance and can satisfy the real-time detection task of infrared targets.

     

/

返回文章
返回