关键词: 交通标志检测, YOLO检测算法, 嵌套残差网络, 多尺度预测, 长沙理工大学, 长沙理工大学中国交通标志检测数据集

Abstract: Automatic driving technology has high requirements for real-time and robustness of traffic sign detection in real world. The YOLOv3-tiny model is a lightweight network with good real-time performance in the object detection, but its accuracy is not high. In this paper, we use YOLOv3-tiny as the basic network and propose a fast traffic sign detection algorithm with three-scale nested residual structure. Firstly, shortcut based on pixel by pixel addition is employed in the YOLOv3-tiny network. It does not increase the number of feature map channels, and a small residual structure is formed in the network at the same time. Secondly, the predictive output with higher spatial resolution is also added through the shortcut, which contains more abundant spatial information, thus forming a large residual structure. Finally, the two residual structures are nested to form a three-scale predictive nested residual network, which makes the main network of Tiny located in these two residual structures and the parameters can be adjusted three times. The results show that the proposed algorithm can quickly and robustly detect traffic signs in real scenes. The F\-1 value of total traffic signs achieves 91.77% on German traffic sign detection benchmark and the detection time is 5ms. On CSUST Chinese traffic sign detection benchmark, F\-1 values of the Mandatory, the Prohibitory and the Warning are 92.41%, 93.91% and 92.03% respectively, and the detection time is 5ms.

Key words: traffic sign detection, you only look once (YOLO) detection algorithm, nested residual network, multi-scale prediction, Changsha University of Science and Technology (CSUST), CSUST Chinese traffic sign detection benchmark (CCTSDB)