1. Real-time Signal Light Detection based on Yolov5 for Railway
- Author
-
Zhangyu Wang, Songyue Yang, Ziren Gong, Bin Zhou, and Wentao Liu
- Subjects
Signal light ,Transformation (function) ,Light detection ,Computer science ,Real-time computing ,Time signal ,Average recall ,Detection theory ,Railway signal - 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.
- Published
- 2021