Skip to main page content
U.S. flag

An official website of the United States government

Dot gov

The .gov means it’s official.
Federal government websites often end in .gov or .mil. Before sharing sensitive information, make sure you’re on a federal government site.

Https

The site is secure.
The https:// ensures that you are connecting to the official website and that any information you provide is encrypted and transmitted securely.

Access keys NCBI Homepage MyNCBI Homepage Main Content Main Navigation
. 2025 Feb 20;12(1):309.
doi: 10.1038/s41597-025-04557-0.

FaultSeg: A Dataset for Train Wheel Defect Detection

Muhammad Zakir Shaikh  1   2   3 , Sahil Jatoi  4   5 , Enrique Nava Baro  6 , Bhagwan Das  7 , Samreen Hussain  8 , Bhawani Shankar Chowdhry  4   5
Affiliations

FaultSeg: A Dataset for Train Wheel Defect Detection

Muhammad Zakir Shaikh et al. Sci Data. .

Abstract

Wheels are a critical component of railway infrastructure and work as the load carrier of the train. However, defective wheels pose a serious risk to safety that can gravely jeopardize people's safety. There is a significant risk of injury or death from defective wheels, endangering the lives of individuals. In this research, FaultSeg dataset is presented for automatic train wheel defect detection for railway transportation around the world. The FaultSeg consists of 829 manually annotated images of faulty wheels acquired by an indigenously developed wayside data acquisition system. Expert Annotators have manually annotated three classes of potential defects: Cracks/Scratches, Shelling, and Discoloration. To assess the practicality of the FaultSeg dataset for training and testing advanced deep learning (DL) models, the dataset was used to train and evaluate the YOLOv9 instance segmentation algorithm. The model achieves an approximate score of 87% accuracy. These results showcase the usability of the FaultSeg dataset in automatic inspection systems and data driven predictive maintenance strategies to safeguard and ensure the safety of railway transportation.

PubMed Disclaimer

Conflict of interest statement

Competing interests: The authors declare no competing interests.

Figures

Fig. 1
Fig. 1
Various Forms of Defect.
Fig. 2
Fig. 2
Yearly Railway Accidents in Pakistan (2019–2023).
Fig. 3
Fig. 3
CAD Model of the Data Acquisition Setup.
Fig. 4
Fig. 4
(a) Deployed Data Acquisition System and (b) Sample Image Captured.
Fig. 5
Fig. 5
Annotated Sample Image using Roboflow.
Fig. 6
Fig. 6
Augmented Image.
Fig. 7
Fig. 7
Project Directory Structure.
Fig. 8
Fig. 8
Confusion Matrix of YOLOv9.
See this image and copyright information in PMC

References

    1. Trepáčová, M., Kurečková, V., Zámečník, P. & Řezáč, P. Advantages and disadvantages of rail transportation as perceived by passengers: A qualitative and quantitative study in the Czech Republic. 10.5507/tots.2020.014.
    1. Alkomy, H. & Shan, J. Modeling and validation of reaction wheel micro-vibrations considering imbalances and bearing disturbances. J Sound Vib492, 115766 (2021).
    1. Magel, E. & Kalousek, J. Identifying and interpreting railway wheel defects. (1996).
    1. Liu, X. Z., Xu, C. & Ni, Y. Q. Wayside Detection of Wheel Minor Defects in High-Speed Trains by a Bayesian Blind Source Separation Method. Sensors19, 3981 (2019). - PMC - PubMed
    1. 537 train accidents reported during last five years - Pakistan Observer. https://pakobserver.net/537-train-accidents-reported-during-last-five-ye....

LinkOut - more resources