A Drone Detection Network Based on Multi-Scale Feature Fusion and Content-Guided Attention Mechanism
To address the limitations of traditional algorithms in detecting small UAV targets, An improved YOLOv8-based UAV detection network is proposed. First, a re-parameterization. RepVGG-style backbone replaces the original YOLOv8 backbone to enable more detailed multi-scale feature extraction and fusion. Second, a content-guided attention fusion module (CGAFusion) is integrated into the neck network, dynamically adjusting the feature weights to enhance the focus on UAV-specific features. Finally, the normalized Wasserstein distance (NWD) is proposed to enhance regression loss optimization and minimize sensitivity to minor target position deviations. Experimental results show that the proposed network achieves a 2.7% improvement in mean average precision (mAP) over compared models.
Authors
- Tao Yao (ORCID: https://orcid.org/0000-0003-2660-1050)
- Lijie Zhang (ORCID: https://orcid.org/0000-0001-5550-9603)
- Yongle Zhang (ORCID: https://orcid.org/0009-0000-6101-7381)
- Saifei Li (ORCID: https://orcid.org/0000-0002-4123-2567)
- Shuo Wang (ORCID: https://orcid.org/0000-0002-2156-5786)
- Yan Li (ORCID: https://orcid.org/0000-0002-6093-3529)
Institutions
- Yantai University (CN)
- Southwestern Institute of Physics (CN)
- Southwest Jiaotong University (CN)
Publication Details
- Published
- 2025-05-23
- DOI
- https://doi.org/10.1109/icaibd64986.2025.11082087
- Primary Topic
- Video Surveillance and Tracking Methods
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China