针对无人机小目标检测难题,提出高效轻量级改进YOLOv11模型。
MASF-YOLO: An Improved YOLOv11 Network for Small Object Detection on Drone View
- 设计多尺度特征聚合模块,提升小目标检测能力。
- 在VisDrone2019上[email protected]提升4.6%,参数减少40%。
- 适合资源受限场景下的高精度无人机目标检测应用。
随着无人机与计算机视觉技术的快速发展,从无人机视角进行目标检测已成为研究热点。然而,目标像素占比极小、尺度变化大及背景复杂等问题严重制约了实际应用。为此,本文基于YOLOv11提出MASF-YOLO网络,包含三个创新模块:多尺度特征聚合模块(MFAM)通过并行多尺度卷积与特征融合显著提升小目标检测精度;改进的高效多尺度注意力模块(IEMA)通过特征分组与跨空间学习增强对目标区域的关注;维度感知选择性融合模块(DASI)自适应加权融合高低维特征。在VisDrone2019数据集上的实验表明,相比YOLOv11-s,MASF-YOLO-s在[email protected]上提升4.6%,[email protected]:0.95提升3.5%;且仅需约60%参数量与65%计算成本,即优于YOLOv11-m。与现有先进方法对比,MASF-YOLO-s在精度与效率上均具明显优势。
原文摘要 · Abstract (English)
With the rapid advancement of Unmanned Aerial Vehicle (UAV) and computer vision technologies, object detection from UAV perspectives has emerged as a prominent research area. However, challenges for detection brought by the extremely small proportion of target pixels, significant scale variations of objects, and complex background information in UAV images have greatly limited the practical applications of UAV. To address these challenges, we propose a novel object detection network Multi-scale Context Aggregation and Scale-adaptive Fusion YOLO (MASF-YOLO), which is developed based on YOLOv11. Firstly, to tackle the difficulty of detecting small objects in UAV images, we design a Multi-scale Feature Aggregation Module (MFAM), which significantly improves the detection accuracy of small objects through parallel multi-scale convolutions and feature fusion. Secondly, to mitigate the interference of background noise, we propose an Improved Efficient Multi-scale Attention Module (IEMA), which enhances the focus on target regions through feature grouping, parallel sub-networks, and cross-spatial learning. Thirdly, we introduce a Dimension-Aware Selective Integration Module (DASI), which further enhances multi-scale feature fusion capabilities by adaptively weighting and fusing low-dimensional features and high-dimensional features. Finally, we conducted extensive performance evaluations of our proposed method on the VisDrone2019 dataset. Compared to YOLOv11-s, MASFYOLO-s achieves improvements of 4.6% in [email protected] and 3.5% in [email protected]:0.95 on the VisDrone2019 validation set. Remarkably, MASF-YOLO-s outperforms YOLOv11-m while requiring only approximately 60% of its parameters and 65% of its computational cost. Furthermore, comparative experiments with state-of-the-art detectors confirm that MASF-YOLO-s maintains a clear competitive advantage in both detection accuracy and model efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。