Rethinking Adaptive Contextual Information and Multi-Scale Feature Fusion for Small-Object Detection in UAV Imagery
Small object detection in unmanned aerial vehicle (UAV) imagery poses significant challenges due to insufficient feature representation, complex background interference, and extremely small target sizes. These factors collectively degrade the performance of conventional detection algorithms, leading to low accuracy, frequent missed detections, and false alarms. To address these issues, we propose <b>YOLO-DMF</b>, which is a novel detection framework specifically designed for drone-based scenarios. Our approach introduces three key innovations from the perspectives of feature extraction and information fusion: (1) a <b>D</b>etail-Semantic Adaptive Fusion (DSAF) module that employs a multi-branch architecture to synergistically enhance shallow detail features and deep semantic information, thereby significantly improving feature representation for small objects; (2) a <b>M</b>ulti-Scale Residual Spatial Attention (MSRSA) mechanism incorporating scale-adaptive spatial attention to improve robustness against background clutter while enabling a more precise localization of critical target regions; and (3) a <b>F</b>eature Pyramid Reuse and Fusion Network (FPRFN) that introduces a dedicated 160×160 detection head and hierarchically combines multi-level shallow features with high-level semantic information through cross-scale fusion, effectively enhancing sensitivity to both small and tiny objects. Comprehensive experiments on the VisDrone2019 dataset demonstrate that YOLO-DMF outperforms state-of-the-art lightweight detection models. Compared to the baseline YOLOv8s, our method achieves improvements of 3.9% in [email protected] and 2.5% in [email protected]:0.95 while reducing model parameters by 66.67% with only a 2.81% increase in computational cost. The model achieves a real-time inference speed of 34.1 FPS on the RK3588 NPU, satisfying the latency requirements for real-time object detection. Additional validation on both the AI-TOD and WAID datasets confirms the method's strong generalization capability and promising potential for practical engineering applications.
Authors
- Qiang Cao (ORCID: https://orcid.org/0000-0001-9890-323X)
- Chang Liu (ORCID: https://orcid.org/0000-0002-7386-1165)
- Yong Wang (ORCID: https://orcid.org/0000-0003-4463-2088)
- Anyu Cheng (ORCID: https://orcid.org/0009-0005-8861-2793)
- Changlei Zhang
Institutions
- Chongqing University of Posts and Telecommunications (CN)
- Chongqing University of Technology (CN)
Publication Details
- Journal
- Sensors
- Published
- 2025-12-01
- DOI
- https://doi.org/10.3390/s25237312
- Primary Topic
- Advanced Neural Network Applications
- Type
- article
- Field-Weighted Citation Impact
- 0.00