改进YOLOv11n的红外船舶检测算法

Improved Infrared Ship Detection Algorithm Based on YOLOv11n

  • 摘要: 针对在红外船舶检测过程中目标尺度差异大、复杂背景干扰导致目标定位不准、误检和漏检的问题,提出一种改进YOLOv11n的红外船舶检测算法。首先,设计C3k2_PMFEM模块作为主干网络的特征提取模块,增强模型的多尺度特征表达能力;其次,利用高效多尺度注意力机制EMA将C2PSA模块改进为C2EMA,通过捕捉通道和空间维度的长距离依赖关系,提升在复杂背景中对目标区域的关注度;再次,基于细节增强卷积提出一种共享细节增强卷积检测头SDCD,加强不同尺度特征信息的交互,提升对图像边缘信息的感知能力,并减少参数冗余;最后,使用Wise-ShapeIoU作为边界框损失函数,减少低质量样本带来的负面梯度,提升边界框回归精度。在红外船舶数据集上的实验结果表明,改进算法在mAP50和mAP50-95上较原算法分别提高了2.5%和2.1%,参数量比原算法降低了12.4%,在检测精度和参数量之间取得了较好平衡。

     

    Abstract: In order to solve the problems of inaccurate target positioning, false detection and missed detection caused by large target scale difference and complex background interference in the process of infrared ship detection, an improved infrared ship detection algorithm of YOLOv11n was proposed. Firstly, the C3k2_PMFEM module was designed as the feature extraction module of the backbone network to enhance the multi-scale feature expression ability of the model. Secondly, the C2PSA module was improved to C2EMA by using the efficient multi-scale attention mechanism EMA, and the attention to the target region in the complex background was improved by capturing the long-distance dependence of the channel and spatial dimensions. Thirdly, based on the detail enhanced convolution, a shared detail enhanced convolutional detection head SDCD was proposed to strengthen the interaction of feature information at different scales, improve the perception of image edge information, and reduce parameter redundancy. Finally, Wise-ShapeIoU is used as the bounding box loss function to reduce the negative gradient caused by low-quality samples and improve the regression accuracy of the bounding box. Experimental results on the infrared ship dataset show that the improved algorithm is 2.5% and 2.1% higher than the original algorithm on mAP50 and mAP50-95, respectively, and the number of parameters is reduced by 12.4% compared with the original algorithm, which achieves a good balance between detection accuracy and parameter quantity.

     

/

返回文章
返回