This article aims to improve the performance of deep learning-based defective fastener inspection method. Due to the defective fasteners are insufficient and far less than the defect-free ones in real railway, it is difficult to train a robust fastener inspection model on such imbalanced dataset. In view of this problem, a novel image generation method called four-discriminator cycle-consistent adversarial network (FD-Cycle-GAN) is proposed to generate the defect fastener images using a large number of defect-free ones. Extensive experiments are conducted on the real fastener images and generated images. Experimental results demonstrate that the defect fastener images generated by our proposed method have better quality and richer diversity than those generated by other state-of-the-art methods. In addition, compared with the CNN-only baseline, the performance of the fastener inspection model trained on the expanded dataset containing the defect fastener images generated by FD-Cycle-GAN is improved significantly. The detection accuracy and relative IMP reach 93.25% and 21.59% respectively.
Four Discriminator Cycle-Consistent Adversarial Network for Improving Railway Defective Fastener Inspection
IEEE Transactions on Intelligent Transportation Systems ; 23 , 8 ; 10636-10645
01.08.2022
5753525 byte
Aufsatz (Zeitschrift)
Elektronische Ressource
Englisch
Low-Cost Fastener Inspection System on High-Speed Railway
Trans Tech Publications | 2011
|High Speed Fastener Inspection
British Library Conference Proceedings | 2016
|