A Comparative Study between State-of-the-Art Object Detectors for Traffic Light Detection

Abstract

In this paper, we evaluate some commonly used state-of-the-art object detectors, namely, Faster RCNN and YOLO, for traffic light detection. We choose the Bosch Small Traffic Light Dataset which is considered to be a challenging benchmark for this purpose. The model architecture, parameters are discussed in detail, and are altered to detect and classify even traffic lights that are indistinguishable to the human eye. We present the results for these optimized models along with the baseline results. Our experimental study shows that Faster RCNN model outperforms YOLO in terms of the precision obtained. However, when real-time deployment is considered, YOLO performs the best.