基于PNL-YOLO的轻量级无人机红外小目标检测

Lightweight Infrared Small Target Detection for UAVs Based on PNL-YOLO

  • 摘要: 针对无人机航拍的红外图像中目标尺寸小、信噪比低、背景复杂且易受噪声干扰等问题,本文以YOLOv11n为基础架构,提出了一种改进的轻量级检测模型PNL-YOLO。在特征提取阶段引入部分卷积以降低计算量并保持模型的特征表达能力;新增小目标检测层并移除大目标检测层以优化多尺度检测能力,同时对检测头进行轻量化重设计,降低计算复杂度;此外,采用归一化Wasserstein距离(NWD)作为相似性度量,有效解决了小目标位置偏差敏感性问题,提升检测精度。实验结果表明,在HIT-UAV数据集上,本模型与YOLOv11n原模型相比,平均检测精度(mAP50)提升了6.08%,参数量和计算量分别减少了42.24%和17.46%,且模型体积大小仅为3.2 MB。综上所述,该方法能够在保持实时性的同时显著提升小目标的检测性能,为无人机红外复杂场景下的目标检测任务提供有效解决方案。

     

    Abstract: To address the issues of small target size, low signal-to-noise ratio, complex background, and susceptibility to noise in unmanned aerial vehicle (UAVs) infrared images, we propose an improved lightweight detection model, PNL-YOLO (PConv NWD Lite -You Only Look Once), based on the YOLOv11n architecture. In the feature extraction stage, a partial convolution was introduced to reduce the computational load while maintaining the feature expression ability of the model. A small-target detection layer is added, and the large-target detection layer is removed to optimize the multiscale detection capability. Meanwhile, the detection head was redesigned in a lightweight manner to reduce computational complexity. Additionally, the normalized Wasserstein distance (NWD) was adopted as the similarity metric, effectively solving the problem of sensitivity to the position deviation of small targets and improving the detection accuracy. The experimental results showed that, on the HIT-UAV dataset, compared with the original YOLOv11n model, the average detection accuracy (mAP50) of this model increased by 6.08%, the number of parameters and computational load decreased by 42.24% and 17.46%, respectively, and the model size was only 3.2MB. In summary, this method can significantly improve the detection performance of small targets while maintaining real-time performance, to provide an effective solution for target-detection tasks in complex infrared scenes captured by UAVs.

     

/

返回文章
返回