Rail surface defect detection based on deep learning

Abstract

In order to ensure the safety of rail transit, detecting the flaws on the rail surface is vitally important. Instead of present manual inspections, detecting defects on rail surface by an automatic approach enables the work more efficient and safe currently. In this paper, we propose a novel two-stage pipeline method for defect detection on rail surface by localizing rails and sliding a deep convolutional neural network (DCNN) on rail surface. Specifically, in the first stage, we use an anchor-free detector to locate the tracks in original images and get the cropped images which focus on rail part. In the second stage, a trained deep convolutional neural network slide on the cropped images to detect defects and we can finally get the types and approximate locations of the defects on rail surface. The experimental results show that the proposed method has robustness and achieves practical performance in defect detection precision.