PAMF-DETR: A multi-scale-aware transformer for small object detection in UAV imagery
摘要
The rapid advancement and widespread adoption of unmanned aerial vehicles (UAVs) have spurred growing interest in aerial object detection. However, UAV-captured imagery presents significant challenges, including scale variations, cluttered backgrounds, object occlusions, and especially the presence of small objects. While transformer-based models like DETR show promise, they often fall short in extracting fine-grained features crucial for detecting small and densely packed objects. To overcome these challenges, we propose Polarized Attention and Multi-scale Fusion DEtection TRansformer (PAMF-DETR). PAMF-DETR features a lightweight backbone equipped with a Multi-Scale Feature Aggregation (MSFA) block to effectively extract fine-grained features of small objects. We propose a Scale-Compensated Cross-Channel Feature Fusion (SC-CCFF) structure that compensates for scale variations and enhances global context modeling by integrating dual-domain attention with large-kernel convolutions. Additionally, a polarity-aware attention mechanism is embedded into the hybrid encoder to improve spatial edge representation. For robust small-object localization, we adopt Inner-MPDIoU, which mitigates background interference during bounding box regression. Extensive experiments on the VisDrone dataset demonstrate that PAMF-DETR outperforms existing mainstream detectors in complex aerial scenarios, achieves a 1.9% improvement in AP and a 3.5% gain in