DF-DETR: An Algorithm for Small Object Detection in UAV Aerial Images
摘要
To address the limitations in small object detection, complex scene understanding, and multi-scale object detection for UAVs, we propose a Dynamic Full-domain Feature Extraction Network (DF-DETR), an enhanced version optimized based on RT-DETR. Firstly, a Convolutional Dynamic Integration Module based on CSPNet (CDIMB) is introduced into the backbone. This module facilitates adaptive deep convolution and efficient feature fusion. Secondly, we introduce and design a Full-domain Transformer to enhance the AIFI module. Leveraging the Transformer architecture’s heightened sensitivity to low-frequency information, this modification significantly improves feature extraction capabilities. Furthermore, a Global-Local Spatial Attention Mechanism (GLSA) is embedded into the P3, P4, and P5 layers to strengthen attention to critical regions and suppress irrelevant background interference. On the VisDrone2019 dataset, the enhanced model achieves a 1.73% improvement in mAP50, while reducing the number of parameters by 22% and computational cost by 6.2%. Compared with other models of similar complexity, the proposed method demonstrates superior performance, validating the effectiveness of these enhancements.