RailNet: An Information Aggregation Network for Rail Track Segmentation

As the basis of scenes understanding for the track inspection task, track segmentation is challenging due to the various illumination conditions, track crossing, and plant coverage. Since the rail has a strong shape prior, strict rail spacing and special distribution in the image, making full use of the spatial information of the rail features becomes an important factor to improve the accuracy of rail segmentation. In this paper, an information aggregation module is proposed to enhance the spatial relationship between pixels of the rail features. In other words, this module expands the receptive field. Furthermore, we build an information aggregation network based on this module, which is called as RailNet. Finally, the RailNet is evaluated in an open train track dataset. Experimental results show that RailNet can achieve the best performance so far in the dataset of trains.

[1]  Garrison W. Cottrell,et al.  Understanding Convolution for Semantic Segmentation , 2017, 2018 IEEE Winter Conference on Applications of Computer Vision (WACV).

[2]  Dinggang Shen,et al.  Lane detection and tracking using B-Snake , 2004, Image Vis. Comput..

[3]  Abhinav Gupta,et al.  Non-local Neural Networks , 2017, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition.

[4]  Haoran Li,et al.  An Efficient Network for Lane Segmentation , 2018, ICCSIP.

[5]  Yi Zhang,et al.  PSANet: Point-wise Spatial Attention Network for Scene Parsing , 2018, ECCV.

[6]  Dongbin Zhao,et al.  Graph-FCN for Image Semantic Segmentation , 2019, ISNN.

[7]  In So Kweon,et al.  VPGNet: Vanishing Point Guided Network for Lane and Road Marking Detection and Recognition , 2017, 2017 IEEE International Conference on Computer Vision (ICCV).

[8]  Andrew Zisserman,et al.  Very Deep Convolutional Networks for Large-Scale Image Recognition , 2014, ICLR.

[9]  Ross B. Girshick,et al.  Focal Loss for Dense Object Detection , 2017, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[10]  Haoran Li,et al.  A temporal-based deep learning method for multiple objects detection in autonomous driving , 2018, 2018 International Joint Conference on Neural Networks (IJCNN).

[11]  Qichao Zhang,et al.  DeepSign: Deep Learning based Traffic Sign Recognition , 2018, 2018 International Joint Conference on Neural Networks (IJCNN).

[12]  Li Liu,et al.  Defect Detection of Rail Surface with Deep Convolutional Neural Networks , 2018, 2018 13th World Congress on Intelligent Control and Automation (WCICA).

[13]  Archana Singh,et al.  Vision based rail track extraction and monitoring through drone imagery , 2017, ICT Express.

[14]  Joonki Paik,et al.  Intelligent Image-Based Railway Inspection System Using Deep Learning-Based Object Detection and Weber Contrast-Based Image Comparison , 2019, Sensors.

[15]  Qichao Zhang,et al.  Reinforcement Learning and Deep Learning based Lateral Control for Autonomous Driving , 2018, IEEE Comput. Intell. Mag..

[16]  Ali Javed,et al.  A Novel Algorithm for Railway Tracks Detection using Satellite Imagery , 2013 .

[17]  Qichao Zhang,et al.  Multi-task learning for dangerous object detection in autonomous driving , 2017, Inf. Sci..

[18]  Shruti Mittal,et al.  Vision Based Railway Track Monitoring using Deep Learning , 2017, ArXiv.

[19]  Weiqiang Ren,et al.  LaneNet: Real-Time Lane Detection Networks for Autonomous Driving , 2018, ArXiv.

[20]  Oliver Zendel,et al.  RailSem19: A Dataset for Semantic Rail Scene Understanding , 2019, 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW).

[21]  Andreas Uhl,et al.  Track Detection for Autonomous Trains , 2010, ISVC.

[22]  Xiaogang Wang,et al.  Spatial As Deep: Spatial CNN for Traffic Scene Understanding , 2017, AAAI.

[23]  Bart De Schutter,et al.  Deep convolutional neural networks for detection of rail surface defects , 2016, 2016 International Joint Conference on Neural Networks (IJCNN).

[24]  Mehmet Karakose,et al.  A New Computer Vision Based Method for Rail Track Detection and Fault Diagnosis in Railways , 2017 .

[25]  Mohamed Aly,et al.  Real time detection of lane markers in urban streets , 2008, 2008 IEEE Intelligent Vehicles Symposium.

[26]  Qichao Zhang,et al.  Reinforcement Learning and Deep Learning Based Lateral Control for Autonomous Driving [Application Notes] , 2019, IEEE Comput. Intell. Mag..

[27]  Enhua Wu,et al.  Squeeze-and-Excitation Networks , 2017, IEEE Transactions on Pattern Analysis and Machine Intelligence.