Real-time Signal Light Detection based on Yolov5 for Railway

Abstract

To improve the safety and efficiency of train operation, autonomous driving train have developed rapidly in recent years. Among them, the signal detection is one of the most basic functions. However, due to the small size of signal light and the complicated of the railway environment, the signal detection is still a huge problem. The existing methods, such as the approach based on Hough circle transformation, are hard to meet the practical application requirements. In this paper, a real time railway signal lights detection based on Yolov5 is introduced. And a lot of experiments were conducted to prove the effectiveness of the proposed method. The experimental results show that the proposed method achieved 0.972 for both average recall rate and average accuracy rate. Besides, the detection speed of the proposed method reached astonishing 100FPS. Overall, the detection speed and accuracy both meet the practical application requirements.