CSM-YOLO:红外弱小目标检测算法

CSM-YOLO: Infrared Weak Small Target Detection Algorithm

  • 摘要: 针对红外图像中的弱小目标像素较小且背景复杂,导致在红外图像小目标检测过程中存在性能低、漏检率和误检率高的问题,提出了一种基于YOLOv12n的红外弱小目标检测改进算法CSM-YOLO。首先,设计CA-DEConv模块替换主干网中C3k2模块,提高主干网对弱小目标的边缘和轮廓信息的提取能力。其次,引入SCSA注意力机制,使算法动态聚焦在小目标特征的重要区域。最后,设计多尺度特征融合模块对小目标不同尺度的特征进行自适应融合,降低深层网络中小目标信息丢失的概率,提高对小目标的检测性能。在SIRST、SIRST V2和NUDT-SIRST公开数据集上的实验表明,改进算法CSM-YOLO较基线模型YOLOv12n在mAP50上分别提高3%、13.3%和13.2%,在mAP50:95上分别提高0.8%、5.9%和15.5%,参数量降低3.5%。

     

    Abstract: To address the issues of low detection performance, high missed detection rate, and false alarm rate in an infrared small-target detection caused by the small pixel size of weak targets and complex background interference, this study proposes an improved infrared dim and small target detection algorithm based on YOLOv12n, named CSM-YOLO. First, a CA-DEConv module was designed to replace the C3k2 module in the backbone network, enhancing the extraction capability of edge and contour information for weak and small targets. Second, the SCSA attention mechanism was introduced to enable the model to focus dynamically on key regions of small target features. Finally, a multiscale feature fusion module was designed to adaptively fuse features across different scales, reducing the loss of small-target information in deeper network layers and improving the overall detection performance. Experimental results on the public datasets SIRST, SIRST V2, and NUDT-SIRST demonstrate that the proposed CSM-YOLO improves mAP50 by 3%, 13.3%, and 13.2%, respectively, and mAP50:95 by 0.8%, 5.9%, and 15.5%, respectively, compared with the baseline YOLOv12n model, while also reducing the total number of parameters by 3.5%.

     

/

返回文章
返回