EHDC-YOLO: Enhancing Object Detection for UAV Imagery via Multi-Scale Edge and Detail Capture

With the rapid expansion of drone applications, accurate detection of objects in aerial imagery has become crucial for intelligent transportation, urban management, and emergency rescue missions. However, existing methods face numerous challenges in practical deployment, including scale variation handling, feature degradation, and complex backgrounds. To address these issues, we propose Edge-enhanced and Detail-Capturing You Only Look Once (EHDC-YOLO), a novel framework for object detection in Unmanned Aerial Vehicle (UAV) imagery. Based on the You Only Look Once version 11 nano (YOLOv11n) baseline, EHDC-YOLO systematically introduces several architectural enhancements: (1) a Multi-Scale Edge Enhancement (MSEE) module that leverages multi-scale pooling and edge information to enhance boundary feature extraction; (2) an Enhanced Feature Pyramid Network (EFPN) that integrates P2-level features with Cross Stage Partial (CSP) structures and OmniKernel convolutions for better fine-grained representation; and (3) Dynamic Head (DyHead) with multi-dimensional attention mechanisms for enhanced cross-scale modeling and perspective adaptability. Comprehensive experiments on the Vision meets Drones for Detection (VisDrone-DET) 2019 dataset demonstrate that EHDC-YOLO achieves significant improvements, increasing mean Average Precision (mAP)@0.5 from 33.2% to 46.1% (an absolute improvement of 12.9 percentage points) and [email protected]:0.95 from 19.5% to 28.0% (an absolute improvement of 8.5 percentage points) compared with the YOLOv11n baseline, while maintaining a reasonable parameter count (2.81 M vs the baseline’s 2.58 M). Further ablation studies confirm the effectiveness of each proposed component, while visualization results highlight EHDC-YOLO’s superior performance in detecting objects and handling occlusions in complex drone scenarios.