Obstacle detection in dangerous railway track areas by a convolutional neural network
Deqiang He, Kai Li, Yanjun Chen, Jian Miao, Xianwang Li, Sheng Shan, Ruochen Ren
Abstract
Abstract The obstacle detection in the dangerous area of railway track is an important research direction in the field of the driverless train. Traditional obstacle detection methods have many issues, such as complicated steps, low detection accuracy, and slow inspection speed. To overcome these defects, a detection network based on deep learning, named Mask R-CNN, was adopted, as described in this paper. The detection network uses the Mask-RCNN model with ResNet101 as its backbone feature extraction network, which has deeper network layers. Therefore, this network has high detection accuracy for small targets. Data from a subway obstacle test was used to train this network. In addition, data augmentation and transfer learning were adopted to improve the efficacy of the training. To improve the detection speed, the technical framework of the detection was also improved. The test results showed that the precision of the Mask-RCNN model with ResNet101 as its backbone feature extraction network reached 95.7% and that it required an average time of 0.18 s. The proposed model shows satisfactory performance when used for obstacle detection in the dangerous area of the railway track, compared with other networks.