DroneNet-Lite:基于改进YOLO11模型的无人机小目标轻量级高速检测器

DroneNet-Lite: A Lightweight High-Speed Detector for Small Objects in Unmanned Aerial Vehicle Based on Improved YOLO11

  • 摘要: 针对无人机图像中小目标检测难度大、计算资源有限和实时性要求高的挑战,提出了一种轻量化的检测模型(DroneNet-Lite)。该模型基于YOLO11模型改进,主要包含3个创新模块:首先,引入P2小目标增强层,提升对小目标的特征提取能力;然后,为了降低计算复杂度,设计深度卷积空间金字塔池化模块(DWSPPF),采用深度卷积替代标准卷积;最后,为解决通道变化导致的特征丢失问题,构建通道融合瓶颈模块(CFB),通过通道变换机制融合高维特征和低维特征。在VisDrone2019-DET数据集上,将DroneNet-Lite模型与YOLOv8、YOLO11、TPH-YOLOv5等检测模型进行了对比实验,并针对DroneNet-Lite模型的各改进模块进行了消融实验。对比实验结果表明:与YOLOv8-n、YOLO11-n模型相比,参数量为1.9 M的DroneNet-Lite-n模型的mAP50值分别提升了2.2%、1.9%;参数量为7.0 M的DroneNet-Lite-s模型的mAP50值达到44.6%,超越TPH-YOLOv5-s等同类模型;参数量为16.9 M的DroneNet-Lite-m模型的mAP50值达到49.6%,比YOLO11-m模型提升3.1%且参数量减少16%。消融实验结果表明P2小目标增强层、DWSPPF模块和CFB模块均能有效提升检测性能:与YOLO11模型相比,引入P2小目标增强层使得模型的mAP50值提升了2.7%、参数量减少了25.3%,DWSPPF模块在精度略高的前提下使模型的参数量减少了8.4%,CFB模块使模型的mAP50值提升了0.2%;与单独引入P2小目标增强层、DWSPPF模块、CFB模块的模型相比,DroneNet-Lite模型的mAP50值分别提升了1.0%、2.9%、3.5%。综上可知DroneNet-Lite模型在保持轻量化的同时显著提升了小目标检测性能,为资源受限的嵌入式设备提供了高效的检测解决方案。

     

    Abstract: To address the challenges of small object detection in Unmanned Aerial Vehicle images, including detection difficulty, limited computational resources, and high real-time needs, a lightweight detection model named DroneNet-Lite is proposed. Based on YOLO11, the model includes three key novel modules. First, a P2 small object enhancement layer is added for small object enhancement. Second, a Depthwise Spatial Pyramind Pooling Fast(DWSPPF) module is designed to reduce computational complexity by replacing standard convolutions with depthwise convolutions. Finally, a Channel Fusion Bottleneck (CFB) module is constructed to prevent feature loss du-ring channel changes. The CFB module fuses high-dimensional and low-dimensional features through channel transformation mechanisms. Comparative experiments were conducted on the VisDrone2019-DET dataset between DroneNet-Lite and detection models including YOLOv8, YOLO11, and TPH-YOLOv5, and ablation experiments were performed on the improvement modules of DroneNet-Lite. Comparative experimental results demonstrated that DroneNet-Lite-n (1.9 M parameters) achieves mAP50 improvements of 2.2% and 1.9% over YOLOv8-n and YOLO11-n, respectively. DroneNet-Lite-s (7.0 M parameters) attains 44.6% mAP50, surpassing TPH-YOLOv5-s and other comparable models. DroneNet-Lite-m (16.9 M parameters) reaches 49.6% mAP50, representing a 3.1% improvement over YOLO11-m while reducing parameters by 16%. Ablation experimental results indicate that the P2 small object enhancement layer, DWSPPF module, and CFB module all effectively enhance detection performance. Compared with YOLO11, introducing the P2 small object enhancement layer raises mAP50 by 2.7% and cuts parameters by 25.3%. The DWSPPF module trims 8.4% of parameters while slightly improving detection accuracy. The CFB module adds another 0.2% mAP50 gain. Compared to individual module integration, DroneNet-Lite achieved mAP50 improvements of 1.0%, 2.9%, and 3.5% for P2 small object enhancement layer, DWSPPF module, and CFB module, respectively. In conclusion, DroneNet-Lite significantly enhances small object detection perfor-mance while maintaining lightweight characteristics, providing an efficient detection solution for resource-constrained embedded devices.

     

/

返回文章
返回