Real-Time Object Detection for the Running Train Based on the Improved YOLO V4 Neural Network

Abstract

Matching the detection speed and accuracy in practical applications is considered to improve the speed of video object detection in front of trains. Lightweight convolutional neural network MobileNet and clustering ideas are combined to improve the object detection algorithm, and the MYOLO-lite model object detection algorithm is designed. The self-made object dataset of a forward-moving train is combined with the K-means clustering idea. The transcendental frame is redesigned to enhance scale adaptability. The nonmaximum suppression and loss function improvement methods of the MYOLO-lite network model are proposed for occluded tracks in front of the train, obstruction of trains, low detection accuracy of large object coincidence, and uneven distribution of positive and negative samples. The mean average precision value of the experimentally designed MYOLO-lite model object detection algorithm reaches 95.74%, with 42.04 frames per second. Detecting a picture only takes 0.024 s.