Material classification and semantic segmentation of railway track images with deep convolutional neural networks

The condition of railway tracks needs to be periodically monitored to ensure passenger safety. Cameras mounted on a moving vehicle such as a hi-rail vehicle or a geometry inspection car can generate large volumes of high resolution images. Extracting accurate information from those images has been challenging due to background clutter in railroad environments. In this paper, we describe a novel approach to visual track inspection using material classification and semantic segmentation with Deep Convolutional Neural Networks (DCNN). We show that DCNNs trained end-to-end for material classification are more accurate than shallow learning machines with hand-engineered features and are more robust to noise. Our approach results in a material classification accuracy of 93.35% using 10 classes of materials. This allows for the detection of crumbling and chipped tie conditions at detection rates of 86.06% and 92.11%, respectively, at a false positive rate of 10 FP/mile on the 85-mile Northeast Corridor (NEC) 2012-2013 concrete tie dataset.

[1]  Matti Pietikäinen,et al.  Multiresolution Gray-Scale and Rotation Invariant Texture Classification with Local Binary Patterns , 2002, IEEE Trans. Pattern Anal. Mach. Intell..

[2]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[3]  Yoshua Bengio,et al.  Gradient-based learning applied to document recognition , 1998, Proc. IEEE.

[4]  Rama Chellappa,et al.  Robust Fastener Detection for Autonomous Visual Railway Track Inspection , 2015, 2015 IEEE Winter Conference on Applications of Computer Vision.

[5]  Ettore Stella,et al.  A Real-Time Visual Inspection System for Railway Maintenance: Automatic Hexagonal-Headed Bolts Detection , 2007, IEEE Transactions on Systems, Man, and Cybernetics, Part C (Applications and Reviews).

[6]  Trevor Darrell,et al.  Rich Feature Hierarchies for Accurate Object Detection and Semantic Segmentation , 2013, 2014 IEEE Conference on Computer Vision and Pattern Recognition.

[7]  Lawrence D. Jackel,et al.  Backpropagation Applied to Handwritten Zip Code Recognition , 1989, Neural Computation.

[8]  Saman A. Zonouz,et al.  Identification Using Encrypted Biometrics , 2013, CAIP.

[9]  Ali Tajaddini,et al.  A Machine Vision System for Automated Joint Bar Inspection From a Moving Rail Vehicle , 2007 .

[10]  Simon Haykin,et al.  GradientBased Learning Applied to Document Recognition , 2001 .

[11]  Sadananda Sahu,et al.  DELAYED ETTRINGITE FORMATION IN SWEDISH CONCRETE RAILROAD TIES , 2004 .

[12]  Trevor Darrell,et al.  Caffe: Convolutional Architecture for Fast Feature Embedding , 2014, ACM Multimedia.

[13]  B. S. Manjunath,et al.  Texture Features for Browsing and Retrieval of Image Data , 1996, IEEE Trans. Pattern Anal. Mach. Intell..

[14]  Dumitru Erhan,et al.  Going deeper with convolutions , 2014, 2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[15]  Trevor Darrell,et al.  Fully Convolutional Networks for Semantic Segmentation , 2017, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[16]  Ettore Stella,et al.  A GPU-based vision system for real time detection of fastening elements in railway inspection , 2009, 2009 16th IEEE International Conference on Image Processing (ICIP).

[17]  Kunihiko Fukushima,et al.  Neocognitron: A self-organizing neural network model for a mechanism of pattern recognition unaffected by shift in position , 1980, Biological Cybernetics.

[18]  David G. Lowe,et al.  Fast Approximate Nearest Neighbors with Automatic Algorithm Configuration , 2009, VISAPP.

[19]  Pavel Babenko,et al.  Visual inspection of railroad tracks. , 2011 .

[20]  Ying Li,et al.  Rail Component Detection, Optimization, and Assessment for Automatic Rail Track Inspection , 2014, IEEE Transactions on Intelligent Transportation Systems.

[21]  Narendra Ahuja,et al.  Automated Visual Inspection of Railroad Tracks , 2013, IEEE Transactions on Intelligent Transportation Systems.

[22]  Michael D.A. Thomas,et al.  The effect of fly ash composition on the expansion of concrete due to alkali-silica reaction , 2000 .

[23]  Matti Pietikäinen,et al.  Rotation Invariant Image Description with Local Binary Pattern Histogram Fourier Features , 2009, SCIA.

[24]  Rama Chellappa,et al.  Discrete shearlet transform on GPU with applications in anomaly detection and denoising , 2014, EURASIP Journal on Advances in Signal Processing.