A Traffic Sign Recognition Method Based on YOLOv5 Deep Learning Algorithm

Abstract

Aiming at the problems of low accuracy and slow recognition efficiency of the traditional traffic sign recognition algorithm in complex environment, a deep learning traffic sign recognition method based on YOLOv5 is proposed. Firstly, the Chinese traffic sign data set TT100K is randomly divided into training set and test set. Convolutional neural network YOLOv4 and convolutional neural network YOLOv5 are used to train respectively on the training set, so as to build the prediction model of traffic signs. Then the trained model is validated on the test set. Through the evaluation of the experimental, it is found that compared with YOLOv4 model, YOLOv5 model has higher recognition accuracy and faster recognition speed.