Automated detection of railway defective fasteners based on YOLOv8-FAM and synthetic data using style transfer

Shi Qiu,Benxin Cai,Weidong Wang,Jin Wang,Qasim Zaheer,Xianhua Liu,Wenbo Hu,Jun Peng
DOI: https://doi.org/10.1016/j.autcon.2024.105363
IF: 10.3
2024-03-13
Automation in Construction
Abstract:Fastener damage detection is an integral component of track safety inspections. The lack of balanced dataset caused by insufficient data on defective fasteners poses a significant challenge to the current development of robust fastener detection models. This study proposes the YOLOv8-FAM detection model algorithm, which combines the enhanced capabilities of YOLOv8, and generates realistic images of defective fasteners by performing style transfer on masked images of non-defective fasteners, thereby creating a balanced dataset for training the detection model. Experimental results show that the defect detection accuracy of YOLOv8-FAM improves by 8% compared to the original model, while also reducing the data acquisition cost by 40%. The YOLOv8-FAM model surpasses existing models in detecting defective fasteners while minimizing inference costs to the maximum extent. The proposed style transfer data synthesis method drives the practical deployment of deep learning, offering an efficient and cost-effective solution for the transportation infrastructure industry.
construction & building technology,engineering, civil
What problem does this paper attempt to address?