UAV image small target detection network via attention mechanism and multi-scale feature fusion
Year: 2025
Type: article
Abstract: Aiming at the challenges of small-target feature extraction and low detection accuracy in object detection caused by dense small targets and large target scale variations in UAV aerial images, this paper proposes an Aerial Small-Target Detection Network with Attention Mechanism and Multi-Scale Feature Fusion (AMSFN). First, a Multi-Scale Dilated Fusion Module (MDFM) is designed to enhance the network's feature extraction capability through the combination of multi-scale feature fusion and attention mechanisms. Second, a Global Grouped Coordinate Attention Module (GGCA) is developed to capture multi-dimensional global information and strengthen the feature representation of small targets. Then, the Normalized Wasserstein Distance (NWD) loss function is combined with the Complete Intersection over Union (CIOU) loss function as the localization regression loss to accelerate network convergence. Finally, an additional target detection layer is introduced, and K-means++ is used to cluster anchor boxes. Experimental results on the VisDrone2019 dataset show that compared with YOLOv7, AMSFN improves mAP0.5 and mAP0.5:0.95 by 5.2% and 4.3%, respectively, significantly enhancing the detection accuracy of small targets.
Institution: Henan University of Science and Technology
Language: English
Subfield: Computer Vision and Pattern Recognition
Field: Computer Science
Domain: Physical Sciences
Open Access status: bronze
Award: 基于进化模糊机制的Web新闻挖掘关键技术
OpenAlex ID: W4414515859