Journal of Optoelectronics · Laser, Volume. 35, Issue 4, 396(2024)

Helmet and license plate detection for cyclists based on improved YOLO v5

XIE Hao1,2, JIA Xiaojun2、*, YU Qingcang1, RAN Erfei1,2, and CHEN Weibiao2,3
Author Affiliations
  • 1[in Chinese]
  • 2[in Chinese]
  • 3[in Chinese]
  • show less

    An improved YOLO v5 model for cyclist helmet and license plate detection is proposed to solve the problems of low accuracy,poor generalization ability and single detection categories in helmet detection.Firstly,the convolutional block attention module (CBAM) is introduced into the backbone network to strengthen the key features of the target region and improve the accuracy of the model. Secondly,by optimizing the multi-scale feature module and adding a detection layer for tiny targets in the prediction end,the detection rate of the network for small targets in dense scenes is enhanced,and the generalization ability of the model is improved.Finally,the model training convergence speed is accelerated and target localization accuracy is improved by optimizing the bounding box regression using efficient intersection over union (EIoU) and by clustering new anchor box sizes using the K-means algorithm in the helmet and license plate dataset created.The experimental results show that the improved YOLO v5 model has achieved an increase in detection accuracy rate of 2.5%,a recall rate increase of 3.3%,and an average precision increase of 3.8%,which makes it more suitable for detecting helmet and license plate targets of cyclists.

    Tools

    Get Citation

    Copy Citation Text

    XIE Hao, JIA Xiaojun, YU Qingcang, RAN Erfei, CHEN Weibiao. Helmet and license plate detection for cyclists based on improved YOLO v5[J]. Journal of Optoelectronics · Laser, 2024, 35(4): 396

    Download Citation

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

    Received: Dec. 15, 2022

    Accepted: --

    Published Online: Sep. 24, 2024

    The Author Email: JIA Xiaojun (xjjiad@sina.com)

    DOI:10.16136/j.joel.2024.04.0846

    Topics