An Improved Light-Weight Traffic Sign Recognition Algorithm Based on YOLOv4-Tiny

Abstract

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 real-time requirements of traffic sign recognition task.