Aiming at the shortcomings of existing anti-UAV systems in computational resources and small target detection accuracy when using visual sensors to detect UAV targets, an improved small UAV target detection algorithm is proposed. Based on YOLOv11, this algorithm introduce an Iterative Attention-fusion Module (IAM) is introduced to realize context-aware multi-scale feature fusion, enhancing the model’s perception of features at different scales. Depth-wise Convolution (DWConv) is employed in the module to achieve lightweight design without changing feature dimensions. The backbone network design based on Dynamic Convolutions significantly optimizes model parameters while reducing model complexity and avoiding the “low FLOPs” trap. Experimental results show that the proposed model improves mAP@0.5 by 5.2% compared to YOLOv11 and by 6.5% compared to YOLOv12 on the Det-Fly dataset, with a 73.2% increase in model parameters but only a 59.4% increase in FLOPs. Additionally, the model is validated on a self-collected dataset, achieving an mAP@0.5 of 92.4%, demonstrating excellent performance. In conclusion, the model improves small target detection accuracy and reduces model complexity through the improved iterative attention-fusion module, and dynamic convolutions, holding broad application prospects and theoretical value in anti-UAV system applications.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Research on Lightweight Algorithm for Small Target Detection of Unmanned Aerial Vehicles Based on Improved YOLO

  • Chenxu Fan,
  • Junyu Wei,
  • Shaojing Su,
  • Siyang Huang,
  • Zongqing Zhao,
  • Tao Ou,
  • Jiangjiang Huang

摘要

Aiming at the shortcomings of existing anti-UAV systems in computational resources and small target detection accuracy when using visual sensors to detect UAV targets, an improved small UAV target detection algorithm is proposed. Based on YOLOv11, this algorithm introduce an Iterative Attention-fusion Module (IAM) is introduced to realize context-aware multi-scale feature fusion, enhancing the model’s perception of features at different scales. Depth-wise Convolution (DWConv) is employed in the module to achieve lightweight design without changing feature dimensions. The backbone network design based on Dynamic Convolutions significantly optimizes model parameters while reducing model complexity and avoiding the “low FLOPs” trap. Experimental results show that the proposed model improves mAP@0.5 by 5.2% compared to YOLOv11 and by 6.5% compared to YOLOv12 on the Det-Fly dataset, with a 73.2% increase in model parameters but only a 59.4% increase in FLOPs. Additionally, the model is validated on a self-collected dataset, achieving an mAP@0.5 of 92.4%, demonstrating excellent performance. In conclusion, the model improves small target detection accuracy and reduces model complexity through the improved iterative attention-fusion module, and dynamic convolutions, holding broad application prospects and theoretical value in anti-UAV system applications.