Aiming at the problems of low detection accuracy and inaccurate positioning accuracy of light-weight network in traffic sign recognition task, an improved light-weight traffic sign recognition algorithm based on YOLOv4-Tiny was proposed. By improving the K-means clustering algorithm, the anchor with appropriate size is generated for the traffic sign data set to improve the detection recall rate and target positioning accuracy. The strategy of large-scale feature map optimization is proposed, which enriches the feature level of the network by using the low-level information, strengthens the representation of the feature information of the small target, and improves the detection accuracy of the long-range small target. In view of the problem of missed detection of high overlapping targets in the post-processing stage of the model, the paper proposes an improved NMS algorithm to screen the prediction box, avoid deleting the prediction results of different targets, and further improve the detection accuracy and recall rate of the target. Experimental results show that, compared with the original YOLOv4-Tiny algorithm, the improved algorithm in traffic sign recognition task based on TT100K dataset, mAP and recall are improved by 5.73% and 7.29% respectively, and FPS value is maintained at about 87 f/s, which meets the accuracy and realtime requirements of traffic sign recognition task.INDEX TERMS traffic sign recognition, YOLOv4-Tiny, clustering algorithm, large scale, improved NMS algorithm
scite is a Brooklyn-based organization that helps researchers better discover and understand research articles through Smart Citations–citations that display the context of the citation and describe whether the article provides supporting or contrasting evidence. scite is used by students and researchers from around the world and is funded in part by the National Science Foundation and the National Institute on Drug Abuse of the National Institutes of Health.