Chinese Journal of Liquid Crystals and Displays, Volume. 38, Issue 5, 680(2023)

Road object detection algorithm based on improved YOLOv5s

Qing ZHOU1,2, Gong-quan TAN1,2、*, Song-lin YIN1,2, Yi-nian LI1,2, and Dan-qin WEI1,2
Author Affiliations
  • 1School of Automation and Information Engineering,Sichuan University of Science & Engineering,Zigong 643000,China
  • 2Artificial Intelligence Key Laboratory of Sichuan Province,Yibin 644000,China
  • show less

    Aiming at the problem that the model parameters of the current mainstream target detection algorithms are too large and cannot be transplanted to mobile devices and applied to assisted driving, this paper proposes an improved YOLOv5s target detection algorithm. Firstly, CSPDarknet, the backbone network of YOLOv5s algorithm, is replaced by MobileNet-V3, a lightweight network model, which solves the problem of large network model and many parameters, reduces the network depth and improves the data inference speed. Secondly, a weighted bidirectional feature pyramid structure Bi-FPN is used to enhance feature extraction, and multi-scale features are integrated to expand the receptive field. Finally, the loss function is optimized and CIoU is used as the boundary box regression loss function to improve the slow convergence speed of the original GIoU model, so that the prediction box is more consistent with the real box, and at the same time reduce the difficulty of network training. Experimental results show that compared with YOLOv5s, SSD, YOLOv3 and YOLOv4_tiny, the mAP of the improved algorithm on KITTI dataset is improved by 4.4, 15.7, 12.4 and 19.6, respectively. Compared with YOLOv5s and lightweight network YOLOv4_tiny, the model size is reduced by 32.4 MB and 21 MB respectively, and the detection speed is improved by 17.6% and 43% respectively. The improved algorithm meets the requirements of small model and high accuracy, and provides a solution for improving detection speed and accuracy of road target detection in assisted driving.

    Tools

    Get Citation

    Copy Citation Text

    Qing ZHOU, Gong-quan TAN, Song-lin YIN, Yi-nian LI, Dan-qin WEI. Road object detection algorithm based on improved YOLOv5s[J]. Chinese Journal of Liquid Crystals and Displays, 2023, 38(5): 680

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Research Articles

    Received: Jul. 30, 2022

    Accepted: --

    Published Online: Jul. 4, 2023

    The Author Email: Gong-quan TAN (tgq77@126.com)

    DOI:10.37188/CJLCD.2022-0257

    Topics