MFDA-YOLO: A multiscale feature fusion and dynamic alignment network for UAV small objects detection

MFDA-YOLO:一种用于无人机小目标检测的多尺度特征融合和动态对齐网络

阅读:1

Abstract

Standard detectors such as YOLOv8 face significant challenges when applied to aerial drone imagery, including extreme scale variations, minute targets, and complex backgrounds. Their generic feature fusion architecture is prone to generating false positives and missing small objects. To address these limitations, we propose an improved MFDA-YOLO model based on YOLOv8. The model introduces an Attention-based Intra-scale Feature Interaction (AIFI) module in the backbone network to enhance high-level feature interactions, improve the adaptation to multi-scale targets, and strengthen feature representation. In the neck network, we design the Drone Image Detection Pyramid (DIDP) network, which integrates a space-to-depth convolution module to efficiently propagate multi-scale features from shallow to deep layers. By introducing an omni-kernel module in the cross-stage partial network for image recovery, DIDP can enhance global contextual awareness and eliminate the computational burden to extend the traditional P2 detection layer. Aiming at the problem of insufficient synergy between localization and classification tasks in the detection head, we design the Dynamic Alignment Detection Head (DADH). DADH can achieve cross-task representation optimization through multi-scale feature interaction learning and a dynamic feature selection mechanism, which significantly reduces model complexity and maintains detection accuracy. In addition, we employ the WIoUv3 loss function to dynamically adjust the focusing coefficients and enhance the model's ability to distinguish small targets. Extensive experimental results demonstrate that MFDA-YOLO outperforms existing state-of-the-art methods such as YOLOv11 and YOLOv13 across the VisDrone2019, HIT-UAV, and NWPU VHR-10 datasets. Particularly on the VisDrone2019 dataset, MFDA-YOLO surpasses the baseline YOLOv8n model, achieving a 4.4 percentage point improvement in mAP0.5 and a 2.7 percentage point increase in mAP0.5:0.95. Furthermore, it reduces parameters by 17.2%, effectively lowering both false negative and false positive rates.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。