ISTD-DINO:复杂环境下空间信息增强的红外小目标无人机检测算法

ISTD - DINO: An Infrared Small - Target UAV Detection Algorithm with Spatial Information Enhancement in Complex Environments

  • 摘要: 低空小型无人机应用渐广的同时也存在威胁安全的情况,红外技术凭其在复杂环境有强适应性的特点广泛用于反无人机系统。本文针对无人机的红外图像检测任务中存在的小目标特征使用不完善、复杂背景下特征不明显的问题,提出了ISTD-DINO红外无人机目标检测模型,提出IFE特征增强方法,采用空间和通道注意力、多分支膨胀卷积充分挖掘红外无人机目标特征的空间信息;通过融合不同层次信息捕捉多尺度特征间长距离依赖关系;同时引入EIOU边界框损失函数,引导模型捕捉小目标细节,增加模型的检测精度。实验结果表明,ISTD-DINO模型的AP0.5达到97.5%,表明ISTD-DINO在红外无人机目标检测任务中具有显著优势。ISTD-DINO在在多种复杂场景中识别红外小目标的任务中表现出色,为维护空域秩序及低空经济健康发展提供了具有一定参考意义的技术方案。

     

    Abstract: Although the applications of low-altitude small unmanned aerial vehicles (UAVs) are expanding, certain operational situations threaten safety. Infrared technology is widely used in anti-UAV systems because of its strong adaptability to complex environments. In this study, the problems of incomplete small-target feature usage and indistinct features in complex backgrounds in the infrared image detection task of UAVs were addressed, and the ISTD-DINO infrared UAV target detection model was proposed. The IFE feature enhancement method was proposed, and spatial and channel attention and multi-branch dilated convolution were adopted to fully mine the spatial information of infrared UAV target features; by fusing information at different levels, long-distance dependencies between multi-scale features were captured; the EIoU bounding box loss function was introduced to improve the detection accuracy of the model during training. Experimental results showed that the AP0.5 of the ISTD-DINO model reached 97.5%, indicating that ISTD-DINO has significant advantages in the infrared UAV target detection task. ISTD-DINO performed well in identifying small infrared targets across various complex scenarios, providing a technical solution with significant reference value for maintaining airspace order and the healthy development of the low-altitude economy.

     

/

返回文章
返回