基于IRLT-YOLO的红外图像无人机目标实时检测研究

Real Time Detection of Infrared Image Drone Targets Based on IRLT-YOLO

  • 摘要: 无人机(Unmanned Aerial Vehicle,UAV)已被广泛应用于各个应用领域,但也出现了越来越多的无人机“黑飞”事件,对公共安全构成了巨大威胁。在反无系统中,红外成像传感器能够在昼夜全天时工作,越来越广泛地应用于无人机检测和监视。本文针对无人机的红外图像检测,提出了IRLT-YOLO(Infrared Lightweighting-YOLO)实时红外无人机目标检测算法,设计轻量化网络,减轻主干网络深度,并在头部采用共享卷积的方法进行轻量化操作,从而减少冗余特征。在保证检测性能的基础上实现实时检测,引入基于归一化瓦瑟斯坦距离(Normalized Wasserstein Distance, NWD)的微小目标检测器,将NWD嵌入到损失函数中,以取代原有的IoU度量。实验结果表明IRLT-YOLO模型的精确率、召回率、mAP@0.5、FLOPs和FPS达到95.4%、85.9%、89.5%、4.9 G和167.0帧/s,与基准模型相比实现了计算精度和速度的双重提升。仿真实验表明IRLT-YOLO模型提高了红外场景下对无人机目标的检测识别能力,在实际部署到边缘设备时能够更快、更好地满足反无人机系统的实时检测应用需求。

     

    Abstract: Unmanned aerial vehicles (UAVs) have been extensively employed across various fields; however, the increasing occurrence of unauthorized UAV "black flight" incidents poses a significant threat to public safety. In anti-UAV systems, infrared imaging sensors that are operational day and night are becoming increasingly prevalent in UAV detection and surveillance. This study addresses infrared image detection for UAVs and proposes a real-time infrared lightweight (IRLT)-YOLO target detection algorithm. In designing lightweight networks to reduce the depth of the backbone network, lightweight operations with a shared convolution in the header are employed, thereby minimizing redundant features. Real-time detection is achieved while preserving detection performance by introducing a tiny target detector–based on the normalized Wasserstein distance (NWD)–embedded in the loss function to replace the original Intersection over Union (IoU) metric. Experimental results indicate that the IRLT-YOLO model achieves precision, recall, mean average precision (mAP)@0.5, floating-point operations per second (FLOPs), and frames per second (FPS) of 95.4%, 85.9%, 89.5%, 4.9G, and 167.0, respectively, representing a dual enhancement in computational accuracy and speed compared to the baseline model. Simulation experiments show that the IRLT-YOLO model enhances the detection and recognition capabilities of UAV targets in infrared scenarios, offering fast and effective real-time detection when deployed on edge devices, thereby meeting the demands of real-time detection applications in anti-UAV systems.

     

/

返回文章
返回