<p>Target detection constitutes a pivotal technology for UAVs to achieve autonomous environmental perception, precise operation, and real-time decision-making. To address the challenges of significant target scale variations, high miss rates, and low detection accuracy in UAV aerial imagery object detection tasks, we propose an end-to-end target detection network featuring multi-scale feature extraction and fusion, designated as EA-DETR. Specifically, we replace the deep residual modules in the backbone network with the proposed Dual-Path Scaled Attention (DPSA) Block. By incorporating a single-head attention mechanism into the feature extraction module and dynamically modulating the contribution of the attention branch, this approach enhances feature representation in the backbone while reducing model parameter complexity. Furthermore, to enhance small target detection capability, we incorporate shallow feature information. During small target feature fusion, we propose a Dual-Path Multi-scale Fusion Module (DPMK) based on channel-wise and spatial attention mechanisms. This effectively strengthens feature representation for small targets while suppressing noise interference in complex backgrounds. Experimental results on the VisDrone2019 dataset demonstrate that EA-DETR achieves improvements of 0.92% in mAP50 and 5.76% in precision compared to the baseline model. These findings validate that EA-DETR enhances detection accuracy while maintaining a low false positive rate. The project page: <a href="https://github.com/qc4059/EA-DETR/tree/master">https://github.com/qc4059/EA-DETR/tree/master</a></p>

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

EA-DETR: an enhanced attention and multi-scale fusion detector for UAV images

  • Qi Cao,
  • Zhigang Lian,
  • Yadi Zhao

摘要

Target detection constitutes a pivotal technology for UAVs to achieve autonomous environmental perception, precise operation, and real-time decision-making. To address the challenges of significant target scale variations, high miss rates, and low detection accuracy in UAV aerial imagery object detection tasks, we propose an end-to-end target detection network featuring multi-scale feature extraction and fusion, designated as EA-DETR. Specifically, we replace the deep residual modules in the backbone network with the proposed Dual-Path Scaled Attention (DPSA) Block. By incorporating a single-head attention mechanism into the feature extraction module and dynamically modulating the contribution of the attention branch, this approach enhances feature representation in the backbone while reducing model parameter complexity. Furthermore, to enhance small target detection capability, we incorporate shallow feature information. During small target feature fusion, we propose a Dual-Path Multi-scale Fusion Module (DPMK) based on channel-wise and spatial attention mechanisms. This effectively strengthens feature representation for small targets while suppressing noise interference in complex backgrounds. Experimental results on the VisDrone2019 dataset demonstrate that EA-DETR achieves improvements of 0.92% in mAP50 and 5.76% in precision compared to the baseline model. These findings validate that EA-DETR enhances detection accuracy while maintaining a low false positive rate. The project page: https://github.com/qc4059/EA-DETR/tree/master