建筑设备科学与工程学院

Signal Image and Video Processing

Small object detection · UAV images · Attention mechanism · Cross-scale feature fusion

Object detection in Unmanned Aerial Vehicle (UAV) images has diverse applications, leading to increasing research interest. Despite the success of detection in natural scenes, UAV images pose two unique challenges: the high prevalence of small objects and significant variations in object scales, limiting the performance of existing methods. To address these, we propose Att-YOLO, a novel small object detection model for UAV images, which improves YOLOv7 with attentive learning and attentional fusion. First, during feature extraction, we introduce an attentive representation learning module with a spatial attention mechanism to highlight foreground features and a channel attention module to reduce background noise. Second, we design an attentional feature fusion strategy to leverage multi-scale feature correlations, assigning dynamic weights to better integrate cross-layer information, which is crucial for handling scale variations. Third, to improve small object detection, we extend the Generalized Efficient Layer Aggregation Network (GELAN) with Swin Transformer blocks, enabling the model to capture both local and global features effectively. Additionally, Wise-IoU (WIoU) v3 is used as the bounding box regression loss to improve localization precision. Extensive experiments show that Att-YOLO achieves competitive performance with state-of-the-art methods, achieving a mean Average Precision (mAP) of 41.8% on the VisDrone2019 dataset and 28.2% on the UAVDT dataset, while maintaining superior AP50 and showing advantageous computational efficiency

fengzengxi

Journal paper

19(14): 1235.

1863-1703

no

2025-10-03