A Multiple Species Railway Defects Detection Method Based on Sample Generation

Abstract

In this paper, we focus on constructing a multi-category railway defect detection method, which are important in both railway operation and railway maintenance. We designed a deep learning based railway defect detection system that include railway classification, switch spacing measurement, railway defect sample expansion based on generative adversarial networks and railway defect detection networks. Deep learning can build fast and accurate defect detection networks, however, its application in railway scenarios is limited due to the scarcity of defect samples, and usually focus on single type defects detection. We build a railway defect detection system by balancing positive and negative samples, contactless switch spacing measurements, generating railway defect samples, and transfer learning. We conduct experiments to show that such an approach can be well applied to real railway detection, which greatly solves the problem of low sample size and low generalisation of deep learning in railway scenarios. Furthermore, due to our method does not have any requirements for the types and scenarios of railways, most deep learning based railway defect detection method can be improved based on our method, which can reduce the difficulty of applying deep learning to railway defect detection. We hope this can promote the research of railway defect detection.

Topics & Concepts