pp. 3755-3771
S&M4150 Research paper of Special Issue https://doi.org/10.18494/SAM5487 Published: August 28, 2025 YOLOv3-based Detection Method for Sensing Railway Fastener Defect with Training Data Generated by Generative Adversarial Network Models [PDF] Ming-An Chung, Chia-Wei Lin, Jia-Wei Lin, Chun-Chia Lin, Chen-You Gao, Pu-Chun Chen, and Yi-Xuan Ma (Received November 22, 2024; Accepted August 14, 2025) Keywords: YOLOv3, railway fastener, defect detection, rail transport, GAN, WGAN
Manual railway inspection, being both time-consuming and labor-intensive, no longer meets the demands of modern railway maintenance, where efficiency and precision are essential. To address this issue, an automatic rail fastener detection system is proposed based on the lightweight You Only Look Once version 3 (YOLOv3)-tiny architecture is proposed. This approach not only leverages the speed advantages of YOLOv3-tiny but also incorporates generative adversarial networks (GANs), along with three of its variants, Wasserstein Generative Adversarial Network (WGAN), Divergence Generative Adversarial Network (WGAN-div), and Wasserstein Generative Adversarial Network with Gradient Penalty (WGAN-GP), to augment the dataset and alleviate the problem of limited defective fastener samples. The quality of the generated images is quantitatively evaluated using peak signal-to-noise ratio (PSNR), structural similarity (SSIM), Fréchet inception distance (FID), and inception score (IS) metrics. These synthetic samples are then integrated into the original dataset to train the YOLO models in a joint learning process. Experimental results show that, after GAN-based augmentation, YOLOv3-tiny, YOLOv3, YOLOv7-tiny, and YOLOv7 achieve mAP0.5 scores of 97.3, 98.7, 98.6, and 98.5% respectively, with particularly significant improvements observed in mAP95. These results demonstrate the effectiveness of the proposed method in addressing data imbalance and enhancing both model accuracy and generalization. In addition, analysis of computational complexity and inference speed indicates that YOLOv3-tiny, with only 14.3 Giga Floating Point Operations (GFLOPs) of computational load, achieves an inference speed of 138.9 Frames Per Second (FPS). This high level of real-time performance, combined with high accuracy, makes YOLOv3-tiny a highly suitable choice for deployment on edge devices in practical railway inspection applications.
Corresponding author: Ming-An Chung![]() ![]() This work is licensed under a Creative Commons Attribution 4.0 International License. Cite this article Ming-An Chung, Chia-Wei Lin, Jia-Wei Lin, Chun-Chia Lin, Chen-You Gao, Pu-Chun Chen, and Yi-Xuan Ma, YOLOv3-based Detection Method for Sensing Railway Fastener Defect with Training Data Generated by Generative Adversarial Network Models, Sens. Mater., Vol. 37, No. 8, 2025, p. 3755-3771. |