Journal of Optoelectronics · Laser, Volume. 35, Issue 4, 396(2024)
Helmet and license plate detection for cyclists based on improved YOLO v5
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.
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
Received: Dec. 15, 2022
Accepted: --
Published Online: Sep. 24, 2024
The Author Email: JIA Xiaojun (xjjiad@sina.com)