Traffic Sign Classification based on Deep Learning

Abstract

In the context of 5G communication, technologies such as the Internet of Things, big data, and supercomputers have been further developed, and artificial intelligence technology has also become one of the hot topics at present. In recent years, the image recognition technology of artificial intelligence has been pursued by most academic enthusiasts and has been rapidly developed. With the development of computer vision, a series of target detection algorithms have emerged in recent years. This paper classifies traffic signs based on the target detection algorithm of YOLOv5 and make use of the advantages of the accuracy and real-time performance of the YOLO algorithm. To restore the real traffic scene, different shooting angles and different shooting locations are prepared as data sets. Finally, to realize the detection and recognition of traffic signs and optimize the algorithm to improve the recognition rate.