改进YOLOv8的多尺度特征协同小目标检测算法OA
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 算法有明显较优的检测效果.
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.
侯汶伶;许滌平;何兆益;林永盛;李圣博
重庆交通大学 交通运输学院,重庆 400074贵州高速公路集团有限公司,贵州 安顺 561000重庆交通大学 交通运输学院,重庆 400074重庆交通大学 交通运输学院,重庆 400074||重庆城市通卡支付有限责任公司,重庆 401147重庆交通大学 交通运输学院,重庆 400074
信息技术与安全科学
YOLOv8小目标检测无人机共享卷积
YOLOv8small target detectiondronesshared convolution
《红外技术》 2026 (9)
1115-1124,10
国家重点研发计划项目(2018YFB1600201)贵州省交通运输厅科技项目(2023-122-023).
评论