基于改进YOLOv11n的无人机航拍图像目标检测算法

Target Detection Algorithm for UAV Aerial Images Based on Improved YOLOv11n

  • 摘要: 针对无人机航拍图像中目标尺寸小、目标密集、漏检和误检等问题,提出了一种改进YOLOv11的小目标检测算法GR-YOLOv11n. 首先引入了一种高效的特征提取与融合模块GhostNet v2替换主干网络中C3k2,通过其独特的深度可分离卷积和注意力机制,能够提取出更为丰富和准确的特征信息,增强了小目标检测模型的特征提取能力. 其次,设计了C2f-RepNCSPFPN模块,它通过结合C2f和RepNCSPFPN的架构优势,优化卷积结构和高效的特征处理流程,显著降低了模型的计算复杂度和参数量,同时保持了较高的检测精度. 在公开数据集VisDrone2019上,本文算法GR-YOLOv11n比YOLOv11n在mAP50上提高了7.6%,mAP50-95上提高了5.4%. 与其他主流目标检测方法相比,改进模型在满足轻量化的同时提高了小目标的检测精度,且满足实时检测需求.

     

    Abstract: In order to solve the problems of small target size, dense targets, missed detection and false detection in UAV aerial images, a small target detection algorithm with improved YOLOv11 was proposed, GR-YOLOv11n. Firstly, an efficient feature extraction and fusion module, GhostNet v2, is introduced to replace C3k2 in the backbone network, which can extract richer and more accurate feature information through its unique deep separable convolution and attention mechanism, and enhance the feature extraction ability of the small object detection model. Secondly, the C2f-RepNCSPFPN module was designed, which optimizes the convolution structure and efficient feature processing process by combining the architectural advantages of C2f and RepNCSPFPN, significantly reduces the computational complexity and parameter quantity of the model, and maintains high detection accuracy. On the public dataset VisDrone2019, The proposed GR-YOLOv11n algorithm achieves a 7.6% improvement in mAP@0.5 and a 5.4% gain in mAP@0.5:0.95 compared to the baseline YOLOv11n. Compared with other mainstream object detection methods, the improved model not only satisfies the lightweight, but also improves the detection accuracy of small targets, and meets the requirements of real-time detection.

     

/

返回文章
返回