ER-YOLO: enhanced information representation network for UAV-based small-object detection
摘要
Object detection in unmanned aerial vehicle (UAV) imagery poses unique challenges due to dense small objects, large scale variations, and complex backgrounds. To address these issues, we propose ER-YOLO, an enhanced variant of YOLO11 tailored for UAV scenarios. ER-YOLO incorporates a Progressive Window Transformer (PWT) module in the backbone to capture both local and global contextual relationships, strengthening feature extraction for small and densely distributed targets. A Bidirectional Efficient Feature Pyramid Network (BEFPN) is further designed to integrate high-resolution and deep semantic features, effectively improving multi-scale representation. Within the BEFPN, a Cross-Spatial Feature Fusion (CSFF) module adaptively calibrates spatial and channel interactions through a dual-path attention mechanism, ensuring efficient and discriminative feature aggregation. Extensive experiments on standard UAV benchmarks demonstrate that ER-YOLO achieves state-of-the-art accuracy while maintaining high inference efficiency. Specifically, ER-YOLOs attains mAP scores of 42.3% on VisDrone and 88.9% on UAVDT, offering a robust solution for UAV object detection.