Real-Time Traffic Sign Recognition using YOLOv3 based Detector

Abstract

Increase in the number of vehicles on road necessitates the use of automated systems for driver assistance. These systems form important components of self-driving vehicles also. Traffic Sign Recognition system is such an automated system which provides the required contextual awareness for the self-driving vehicle. CNN based methods like Faster R-CNN for object detection provide human level accuracy and real time performance and are proven successful in Traffic Sign Recognition systems [1]. Single stage detection systems such as YOLO [2] and SSD [3], despite offering state-of-the-art real-time detection speed, are not preferred for traffic sign detection problem due to its reduced accuracy and small object detection issues. The enhanced YOLO versions, YOLOv2 and YOLOv3 have shown promising results with respect to accuracy and speed required for object detection problems. YOLOv3 uses specialized network architecture inspired from feature pyramid network and has several design changes over previous versions to tackle the low accuracy and small object detection problems. In this paper, an approach for traffic sign recognition system based on YOLOv3 is presented with comparative analysis of its performance with Faster R-CNN based sign detector [1]. YOLOv3 forms the traffic sign detection network and a CNN-based classifier forms the traffic sign class recognizer. The network training and evaluation are done using the German Traffic Sign Detection Benchmark (GTSDB) [5] dataset and the classifier performance is verified using German Traffic Sign Recognition Benchmark (GTSRB) [6] dataset.