FaultSeg: A Dataset for Train Wheel Defect Detection
Authors/Creators
Description
The dataset contains original raw images of train wheels captured using a GoPro Hero 9 Black camera, along with their respective segmentation labels for real-time wheel defect detection. The images are annotated for four distinct classes: Wheel, Shelling, Discoloration, and Cracks/Scratches.
It is pertinent to mention here that the model confuses between following classes: peeling, cracking, and scratches. We have categorised all of the cracks and scratches in our dataset into a single class called cracks/scratches.
Annotated Data:
This data is further divided into formats and stored within three folders: train, test, and valid. The formats include:
— JSON: Located in the “Labeled_data_coco_segmentation_JSON.zip” folder.
— XML: Found in the “Labeled_data_voc_XML.zip” folder.
— TXT: Available in the “Labeled_data_TXT.zip” folder.
— TFRecord: Under the “Labeled_data_tfrecord.zip” folder.
— CSV: Located in the “labeled_data_multiclass_CSV.zip” folder.
These formats strengthen the overall usability of the code by facilitating the training of various AI-based models, including YOLO, Detectron 2, FastInst, and many others.
For detailed annotation of the dataset, please go through this Roboflow link.
Files
Additional details
Related works
- Is part of
- Journal article: 10.1109/ACCESS.2023.3240167 (DOI)
Funding
- Higher Education Commission
- NCRA-CMS Lab 2(1076)/HEC/M&E/2018/704
Dates
- Collected
-
2023-01-25Data Collection
Biodiversity
- Event date
- 2023-01-25 , 2023-01-26
- Individual count
- 5
- Country
- Pakistan
References
- Advantages and disadvantages of rail transportation as perceived by passengers: A qualitative and quantitative study in the Czech Republic | Transactions on Transport Sciences. https://www.tots.upol.cz/artkey/tot-202003-0005_advantages-and-disadvantages-of-rail-transportation-as-perceived-by-passengers-a-qualitative-and-quantitative.php.
- Alkomy, H. & Shan, J. Modeling and validation of reaction wheel micro-vibrations considering imbalances and bearing disturbances. J Sound Vib 492, 115766 (2021).
- Pakistan Observer. 537 train accidents reported during last five years. 0–1 (2024).
- Srinivasarao, G. et al. Deep learning based condition monitoring of road traffic for enhanced transportation routing. Journal of Transportation Security 17, (2024).
- Serradilla, O., Zugasti, E., Rodriguez, J. & Zurutuza, U. Deep learning models for predictive maintenance: a survey, comparison, challenges and prospects. Applied Intelligence 2022 52:10 52, 10934–10964 (2022).
- Jamwal, A., Agrawal, R. & Sharma, M. Deep learning for manufacturing sustainability: Models, applications in Industry 4.0 and implications. International Journal of Information Management Data Insights 2, 100107 (2022).
- Tachtatzis, C. et al. Condition Monitoring and Predictive Maintenance of Assets in Manufacturing Using LSTM-Autoencoders and Transformer Encoders. Sensors 2024, Vol. 24, Page 3215 24, 3215 (2024).
- Skarlatos, D., Karakasis, K. & Trochidis, A. Railway wheel fault diagnosis using a fuzzy-logic method. Applied Acoustics 65, 951–966 (2004).
- Bai, Y., Yang, J., Wang, J. & Li, Q. Intelligent Diagnosis for Railway Wheel Flat Using Frequency-Domain Gramian Angular Field and Transfer Learning Network. IEEE Access 8, 105118–105126 (2020).
- Bernal, E., Spiryagin, M. & Cole, C. Ultra-Low Power Sensor Node for On-Board Railway Wagon Monitoring. IEEE Sens J 20, 15185–15192 (2020).
- Li, H., Wang, H., Xie, Z. & He, M. Fault diagnosis of railway freight car wheelset based on deep belief network and cuckoo search algorithm. https://doi.org/10.1177/09544097211029155 236, 501–510 (2021).
- Fu, W. et al. Recent Advances in Wayside Railway Wheel Flat Detection Techniques: A Review. Sensors 2023, Vol. 23, Page 3916 23, 3916 (2023).
- Arain, A. et al. Railway track surface faults dataset. Data Brief 52, 110050 (2024).
- Shaikh, M. Z. et al. State-of-the-Art Wayside Condition Monitoring Systems for Railway Wheels: A Comprehensive Review. IEEE Access 11, 13257–13279 (2023).
- Zhang, Z., Shao, S. & Gao, Z. A novel method on wheelsets geometric parameters on line based on image processing. 2010 International Conference on Measuring Technology and Mechatronics Automation, ICMTMA 2010 1, 257–260 (2010).
- Karaköse, M., Yaman, O. & Akın, E. Real time implementation for fault diagnosis and condition monitoring approach using image processing in railway switches. International Journal of Applied Mathematics Electronics and Computers 307–313 (2016) doi:10.18100/IJAMEC.270627.
- Tastimur, C., Yaman, O., Karakose, M. & Akin, E. A real time interface for vision inspection of rail components and surface in railways. IDAP 2017 - International Artificial Intelligence and Data Processing Symposium (2017) doi:10.1109/IDAP.2017.8090267.
- Bernal, E., Spiryagin, M. & Cole, C. Wheel flat detectability for Y25 railway freight wagon using vehicle component acceleration signals. Vehicle System Dynamics 58, 1893–1913 (2019).
- Karakose, E., Gencoglu, M. T., Karakose, M., Aydin, I. & Akin, E. A new experimental approach using image processing-based tracking for an efficient fault diagnosis in pantograph-catenary systems. IEEE Trans Industr Inform 13, 635–643 (2017).
- Santur, Y., Karakose, M. & Akin, E. An adaptive fault diagnosis approach using pipeline implementation for railway inspection. Turkish Journal of Electrical Engineering and Computer Sciences 26, 987–998 (2018).
- Choudhary, A. K. & Ahmad Khan, D. Introduction to Conditioning Monitoring of Mechanical Systems. Advances in Intelligent Systems and Computing 1096, 205–230 (2020).
- Sikora, P. et al. Artificial Intelligence-Based Surveillance System for Railway Crossing Traffic. IEEE Sens J 21, 15515–15526 (2021).
- Li, Y., Zuo, M. J., Lin, J. & Liu, J. Fault detection method for railway wheel flat using an adaptive multiscale morphological filter. Mech Syst Signal Process 84, 642–658 (2017).
- Torabi, M., Mohammad Mousavi, S. & Younesian, D. A High Accuracy Imaging and Measurement System for Wheel Diameter Inspection of Railroad Vehicles. IEEE Transactions on Industrial Electronics 65, 8239–8249 (2018).
- Huang, Y., Lin, J., Liu, Z. & Huang, C. A Morphological Filtering Method Based on Particle Swarm Optimization for Railway Vehicle Bearing Fault Diagnosis. Shock and Vibration 2019, 2593973 (2019).
- Kim, G., Kim, H., Koo, J., Author, C. & Seo Koo, J. A Study on Cepstrum Analysis for Wheel Flat Detection in Railway Vehicles. Journal of the Korean Society of Safety 31, 28–33 (2016).
- Shaikh, M. Z., Ahmed, Z., Baro, E. N., Hussain, S. & Milanova, M. Deep learning based identification and tracking of railway bogie parts. Alexandria Engineering Journal 107, 533–546 (2024).
- Soleimani, H., Moavenian, M., Masoudi Nejad, R. & Liu, Z. An applied method for railway wheel profile measurements due to wear using image processing techniques. SN Appl Sci 3, 1–10 (2021).