高级检索

基于改进YOLOv5s的复杂环境道路坑洼检测方法

Road Pothole Detection Method in Complex Environment Based on Improved YOLOv5s

  • 摘要: 为使自动驾驶系统能在复杂环境中有效检测和定位道路坑洼,本文对现有YOLOv5s目标检测算法进行了改进。首先,用MobileNetV3替换原模型的骨干部分,减少模型参数量,轻量化网络模型;其次,在YOLOv5s的颈部引入BiFPN (加权双向特征金字塔网络)模块,在保持模型轻量化设计的同时,提升模型在多尺度特征融合、信息传递、特征表达能力和检测精度等方面的性能;再次,引入生成对抗神经网络的图像风格迁移概念,利用PaddleGAN进行数据扩增,增加数据集的多样性;最后,在自建的数据集上进行模型的训练和验证。实验结果表明:改进的YOLOv5算法的准确率和平均准确率mAP50分别提升了0.035和0.009,检测速度提升至54.1帧/s。改进的算法更轻量化,提高了检测精度,可为复杂环境下坑洼检测提供技术参考。

     

    Abstract: To enable autonomous driving systems to effectively detect and locate road potholes in complex environments, improvements have been made to the existing YOLOv5s object detection algorithm. Firstly, MobileNetV3 is employed to replace the original backbone of the model, reducing the parameter counts and achieving a more lightweight network design. Additionally, a BiFPN (bidirectional feature pyramid network) module is introduced in the neck of YOLOv5s, significantly enhancing the model’s performance in multi-scale feature fusion, information propagation, feature representation, and detection accuracy, while maintaining the lightweight nature of the architecture. Furthermore, the concept of image style transfer from generative adversarial networks is incorporated, utilizing PaddleGAN for data augmentation to increase the diversity of the dataset. Finally, experiments conducted on a custom dataset revealed that the improved YOLOv5s algorithm exhibits a 0.035 enhancement in accuracy and a 0.009 increase in mean average precision mAP50, achieving a detection speed of 54.1 frames per second. The proposed algorithm is more lightweight and enhances detection precision, providing a valuable technical reference for pothole detection in complex environments.

     

/

返回文章
返回