改进YOLOv8的多尺度特征协同小目标检测算法

The Multi-Scale Feature Cooperative Small Target Detection Algorithm of Improved YOLOv8

  • 摘要: 针对当前无人机航拍图像检测存在精度低、漏检率高、模型冗余等问题,本文在YOLOv8的基础上提出改进YOLOv8的多尺度特征协同MSF-YOLO检测算法。首先使用RepViTBlock与EMA注意力机制构造RBE-C2f替换骨干网络中的C2f模块,提升特征提取能力;其次,使用TFE和SSFF以及小目标检测层重新构造颈部网络,实现多特征交互融合;最后,在检测头部分设计了一种共享卷积检测头,使得不同检测头进行特征共享,减少大量冗余特征,提高模型检测效率。改进模型在VisDrone2019数据集上相比于原始模型在精确率、召回率、mAP50、mAP50-95分别提升了3.1、4.0、4.6和2.8个百分点,参数量减少42%,模型大小减少36%,仅为3.8 MB。通过复杂环境的实验对比,改进后的模型相比于基础的YOLOv8算法有明显较优的检测效果。

     

    Abstract: To address problems such as low accuracy, a high missed detection rate, and model redundancy in UAV aerial image detection, an improved multi-scale feature collaborative MSF-YOLO detection algorithm based on YOLOv8 was proposed in this study. First, the RepViTBlock and EMA attention mechanisms were used to construct RBE-C2f to replace the C2f module in the backbone network and improve feature extraction capability. Second, the neck network was reconstructed using the TFE, SSFF, and small-target detection layers to achieve multi-feature interactive fusion. Finally, a shared convolution detection head was designed in the detection head to make different detection heads share features, reduce redundant features, and improve the efficiency of the model detection. Compared with the original model in the VisDrone2019 data set, the accuracy rate, recall rate, mAP50, and mAP50-95 of the improved model increased by 3.1%, 4.0%, 4.6%, and 2.8%, respectively; the number of parameters decreased by 42%, and the model size decreased by 36% to only 3.8 MB. Based on an experimental comparison in a complex environment, the improved model exhibited a better detection effect than the basic YOLOv8 algorithm.

     

/

返回文章
返回