<p>In recent years, the application of Unmanned Aerial Vehicle (UAV) in production activities involving outdoor operations has gradually become widespread and routine. However, challenges such as the wide size variation of small objects, frequent occlusions, and complex backgrounds have emerged as key difficulties in UAV aerial image object detection. This paper proposes PASR-YOLO, an enhanced YOLO11 algorithm optimized for small objects. To address the issue of insufficient representation of small objects in the detection layer, the High Resolution Detail Injection Unit (HDI) is introduced. Building upon the addition of a high resolution detection branch, it innovatively extracts small object features from low level layers and injects them into higher layers. Furthermore, incorporating the Cross Stage Partial network (CSP) based on three different size branch representations, it aggregates multi-scale and fine-grained information, achieving a lightweight and high precision small object feature pyramid. To further reduce the number of parameters, a low-redundancy mapping named n-Scale Recursive Split-Enhance Fusion Block (NSSE) is proposed that accommodates both large and small models, balancing compactness with equivalent feature extraction and gradient flow capabilities. On the VisDrone dataset, PASR-YOLO achieves 6.6% and 4.5% improvements over the baseline model in mAP 0.5 and mAP 0.5–0.95, respectively.</p>

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

PASR-YOLO: a small object detection algorithm for UAV based on cross-scale feature aggregation

  • Tien-Wen Sung,
  • Yujie Lai,
  • Chao-Yang Lee,
  • Shaopei Wen

摘要

In recent years, the application of Unmanned Aerial Vehicle (UAV) in production activities involving outdoor operations has gradually become widespread and routine. However, challenges such as the wide size variation of small objects, frequent occlusions, and complex backgrounds have emerged as key difficulties in UAV aerial image object detection. This paper proposes PASR-YOLO, an enhanced YOLO11 algorithm optimized for small objects. To address the issue of insufficient representation of small objects in the detection layer, the High Resolution Detail Injection Unit (HDI) is introduced. Building upon the addition of a high resolution detection branch, it innovatively extracts small object features from low level layers and injects them into higher layers. Furthermore, incorporating the Cross Stage Partial network (CSP) based on three different size branch representations, it aggregates multi-scale and fine-grained information, achieving a lightweight and high precision small object feature pyramid. To further reduce the number of parameters, a low-redundancy mapping named n-Scale Recursive Split-Enhance Fusion Block (NSSE) is proposed that accommodates both large and small models, balancing compactness with equivalent feature extraction and gradient flow capabilities. On the VisDrone dataset, PASR-YOLO achieves 6.6% and 4.5% improvements over the baseline model in mAP 0.5 and mAP 0.5–0.95, respectively.