面向红外非合作目标的通用增强检测算法

An Enhanced Detection Method for Infrared Non-Cooperative Targets

  • 摘要: 针对机场周边环境下基于热红外图像的非合作目标检测所面临的对比度低、目标分辨率低、背景复杂等挑战,本文提出一种面向红外非合作目标的通用增强检测算法。该算法通过四个核心模块对基准模型进行系统性改进:首先,引入小尺度检测头,显著提升了对微小目标的感知能力与定位精度;其次,采用空间深度转换卷积模块(Space-to-depth convolution,SPDConv)替代传统下采样模块,有效保留了图像的细粒度特征,缓解了细节信息丢失问题;再次,设计了重参数化大核倒瓶颈模块(Reparameterized Large-Kernel Inverted Bottleneck,RepLKIB),采用并行多分支结构提取多尺度特征,重点引入大核卷积分支以扩大感受野;采用倒瓶颈结构,提升特征表征能力;通过结构重参数化技术将训练时的多分支融合为推理时的单路结构,兼顾了精度与效率;最后,采用高效交并比(Efficient Intersection over Union,EIoU)损失替代完全交并比(Complete Intersection over Union,CIoU)损失,通过分离中心与尺度误差项,显著增强低质量样本,尤其是小目标样本的回归梯度,从而提升模型的定位鲁棒性与小目标检出率。在自建的红外非合作目标数据集上的对比实验表明,本文所提算法在基准模型YOLOv10n(YOLOv10的超轻量模型)的基础上,mAP@[0.5:0.95]指标大幅提升了9.1个百分点,达到94.0%。与参数量更大的YOLOv10m(YOLOv10的中等规模模型)相比,所提模型在精度仅下降0.5%的情况下,将参数量与推理时间分别降低至原来的70.3%和75.7%。与YOLO系列其他版本相比,本算法以超轻量模型的推理速度,达到了与中等规模模型相当的精度。同时实验显示基准算法为YOLOv11时,算法同样有效,证明了本算法具有通用性。综上本算法为复杂场景下的红外非合作目标检测提供了一种高效可靠的解决方案。

     

    Abstract: To address the critical challenges of the low contrast and limited resolution of small targets and complex background interference in thermal infrared image-based non-cooperative target detection in airport environments, this paper proposes a universal enhancement detection algorithm specifically optimized for infrared non-cooperative targets. The proposed algorithm systematically improves the baseline model using four core plug-and-play modules. First, a small-scale detection head was introduced to significantly enhance the perception capabilities and localization precision of microscopic targets. Second, a space-to-depth convolution (SPDConv) module was employed to replace the traditional downsampling layers, which effectively preserved the fine-grained image features and mitigated the loss of critical details. Third, a reparameterized large-kernel inverted bottleneck (RepLKIB) module was designed. This module utilizes a parallel multi-branch structure to extract multi-scale features by incorporating a large-kernel convolutional branch to expand the receptive field. Furthermore, an inverted bottleneck structure was adopted to enhance feature representation, whereas structural reparameterization technology was applied to fuse the multi-branch training structure into a single-path inference structure, thereby balancing the accuracy and computational efficiency. Finally, an efficient intersection-over-union loss function was implemented to replace the complete intersection-over-union loss function. By decoupling the center and scale error terms, this approach significantly strengthened the regression gradients for low-quality samples, particularly small targets, thus improving the localization robustness and detection rates.Comparative experiments conducted on a self-constructed comprehensive infrared non-cooperative target dataset demonstrated the superior performance of the proposed algorithm. Building on the YOLOv10n (nano-scale) baseline, the mAP@[0.5:0.95] metric increased substantially by 9.1 percentage points, reaching 94.0%. When compared with the larger YOLOv10m (medium-scale) model, the proposed model reduced the number of parameters and inference time to 70.3% and 75.7% of the original values, respectively, while incurring only a marginal 0.5% decrease in accuracy. A comparison with other versions of the YOLO series indicated that the algorithm achieved an accuracy comparable to that of medium-scale models while maintaining nano-scale inference speeds. Additional experiments using YOLOv11 as the baseline further validated the effectiveness and universality of the proposed modules across different architectures. In conclusion, this algorithm provides an efficient and reliable solution for infrared non-cooperative target detection in complex scenarios.

     

/

返回文章
返回