UAV-YOLO:红外场景下无人机实时目标检测算法

UAV-YOLO: A Real-Time Target Detection Algorithm For UAVs In Infrared Scenes

  • 摘要: 针对红外场景下无人机检测精度低与计算量高的问题,提出一种改进的UAV-YOLOv11n算法。首先,引入加权双向特征金字塔网络(BiFPN),通过优化多尺度特征融合提升模型检测性能;其次,采用轻量级细节增强检测头(LSDECD),在降低参数量的同时增强小目标检测性能;此外,构建卷积注意力融合模块(CAFM)强化特征交互提升鲁棒性;最后,使用Wise-SIoU损失函数以加速模型收敛。实验结果表明,改进模型mAP@50达到91.3%,较YOLOv11n提升1.7%。在公开红外图像弱小飞机目标检测跟踪数据集下验证表明,改进模型各项评价指标均有提升,证明其具有良好的泛化性和鲁棒性。

     

    Abstract: An improved unmanned aerial vehicle (UAV)-YOLO algorithm is proposed to address the problems of low precision and high computation of UAV detection in infrared scenes. First, a weighted bidirectional feature pyramid network (BiFPN) was introduced to enhance the model detection performance by optimizing multi-scale feature fusion. Second, a lightweight shared detail enhanced convolutional detection head (LSDECD) was used to enhance the performance of small-target detection while decreasing the number of parameters, and a convolution and attention fusion module (CAFM) was constructed to strengthen the feature interactions to enhance the robustness. Finally, a Wise–SIoU loss function was used to accelerate model convergence. The experimental results demonstrate that the improved model achieves 91.3% mAP@50, with a 1.7% enhancement compared to the original YOLOv11n. Validation under a weak aircraft target detection and tracking dataset of public infrared images shows that the improved model improves all evaluation indices, proving that it has good generalization and robustness.

     

/

返回文章
返回