微尺度提取与多尺度融合的红外目标检测方法

Micro-scale Extraction and Multi-scale Fusion Method for Infrared Target Detection

  • 摘要: 针对红外小目标检测存在的特征感知能力不足、多尺度特征融合效果欠佳以及边界框定位精度有限等问题,本文提出一种微尺度提取与多尺度融合的红外目标检测方法(CAANet)。以YOLOv8n为基准模型,创新设计了坐标注意力增强的微尺度特征提取模块(CMS Block),通过多尺度深度可分离卷积与坐标注意力机制强化微小目标特征感知;引入自适应空间特征融合模块(ASFF),动态调度多尺度信息以提升融合一致性;设计自适应红外交并比损失函数(AIoU),融合空间几何约束与灰度显著性特征优化定位精度。在IRSTD-1k数据集上的实验表明,CAANet网络的精确率、召回率和mAP@50分别达到80.2%、81.1%和83.8%,显著优于主流模型;在NUDT-SIRST数据集上的鲁棒性测试进一步验证了其泛化能力,可为红外小目标检测任务提供有效的技术支撑。

     

    Abstract: Aiming at the problems of insufficient feature perception, poor multi-scale feature fusion, and limited bounding box localization accuracy in infrared small target detection, this paper proposes an infrared target detection method based on micro-scale extraction and multi-scale fusion (CAANet). Based on YOLOv8n as the baseline model, a Coordinate-Attention Enhanced Micro-Scale Feature Extraction Module (CMS Block) is designed to enhance tiny target feature perception through multi-scale depthwise separable convolutions and a coordinate attention mechanism. An Adaptive Spatial Feature Fusion module (ASFF) is introduced to dynamically schedule multi-scale information and improve fusion consistency. An Adaptive Infrared Intersection over Union (AIoU) loss function is proposed to optimize localization accuracy by integrating spatial geometric constraints with grayscale saliency features. Experimental results on the IRSTD-1k dataset demonstrate that CAANet achieves a precision of 80.2%, a recall of 81.1%, and an mAP@50 of 83.8%, significantly outperforming mainstream detection models. The robustness evaluation on the NUDTSIRST dataset further validates its generalization capability, indicating that the proposed method can provide effective technical support for infrared small target detection tasks.

     

/

返回文章
返回