From Digital Model to Reality Application: A Domain Adaptation Method for Rail Defect Detection

• Published in: Pattern Recognition and Computer Vision Vol. 13020; pp. 115 - 126
• Main Authors: Cui, Wenkai, Wang, Jianzhu, Yu, Haomin, Peng, Wenjuan, Wang, Le, Wang, Shengchun, Dai, Peng, Li, Qingyong
• Format: Book Chapter
• Language: English
• Published: Switzerland Springer International Publishing AG 2021; Springer International Publishing
• Series: Lecture Notes in Computer Science
• Subjects: 3D point cloud; Digital model; Domain adaptation; Object detection; Rail defect detection
• ISBN: 9783030880064; 3030880060
• ISSN: 0302-9743; 1611-3349
• DOI: 10.1007/978-3-030-88007-1_10

Recently, vision-based rail defect detection has attracted much attention owing to its practical significance. However, it still faces some challenges, such as high false alarm rate and poor feature robustness. With the development of deep neural networks (DNNs), deep learning based models have shown the potential to solve the problems. Nevertheless, these models usually require a large number of training samples, while collecting and labeling sufficient defective rail images is somewhat impractical. On the one hand, the probability of defect occurrence is low. On the other hand, we are not able to annotate samples that include all types of defects. To this end, we propose to generate defective training images in the digital space. In order to bridge the gap between virtual and real defective samples, this paper presents a domain adaptation based model for rail defect detection. The proposed method is evaluated on a real-world dataset. Experimental results show that our proposed method is superior to five established baselines.