Open Access iconOpen Access

ARTICLE

crossmark

Online Rail Fastener Detection Based on YOLO Network

Jun Li1, Xinyi Qiu1, Yifei Wei1,*, Mei Song1, Xiaojun Wang2

1 Beijing Key Laboratory of Work Safety Intelligent Monitoring, School of Electronic Engineering, Beijing University of Posts and Telecommunications, Beijing, 100876, China
2 Dublin City University, Dublin, 9, Ireland

* Corresponding Author: Yifei Wei. Email: email

Computers, Materials & Continua 2022, 72(3), 5955-5967. https://doi.org/10.32604/cmc.2022.027947

Abstract

Traveling by high-speed rail and railway transportation have become an important part of people’s life and social production. Track is the basic equipment of railway transportation, and its performance directly affects the service lifetime of railway lines and vehicles. The anomaly detection of rail fasteners is in a priority, while the traditional manual method is extremely inefficient and dangerous to workers. Therefore, this paper introduces efficient computer vision into the railway detection system not only to locate the normal fasteners, but also to recognize the fasteners states. To be more specific, this paper mainly studies the rail fastener detection based on improved You can Only Look Once version 5 (YOLOv5) network, and completes the real-time classification of fastener states. The improved YOLOv5 network proposed contains five sections, which are Input, Backbone, Neck, Head Detector and a read-only Few-shot Example Learning module. The main purpose of this project is to improve the detection precision and shorten the detection time. Ultimately, the rail fastener detection system proposed in this paper is confirmed to be superior to other advanced algorithms. This model achieves on-line fastener detection by completing the “sampling-detection-recognition-warning” cycle of a single sample before the next image is sampled. Specifically, the mean average precision of model reaches 94.6%. And the model proposed reaches the speed of 12 ms per image in the deployment environment of NVIDIA GTX1080Ti GPU.

Keywords


Cite This Article

APA Style
Li, J., Qiu, X., Wei, Y., Song, M., Wang, X. (2022). Online rail fastener detection based on YOLO network. Computers, Materials & Continua, 72(3), 5955-5967. https://doi.org/10.32604/cmc.2022.027947
Vancouver Style
Li J, Qiu X, Wei Y, Song M, Wang X. Online rail fastener detection based on YOLO network. Comput Mater Contin. 2022;72(3):5955-5967 https://doi.org/10.32604/cmc.2022.027947
IEEE Style
J. Li, X. Qiu, Y. Wei, M. Song, and X. Wang "Online Rail Fastener Detection Based on YOLO Network," Comput. Mater. Contin., vol. 72, no. 3, pp. 5955-5967. 2022. https://doi.org/10.32604/cmc.2022.027947



cc This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
  • 1268

    View

  • 782

    Download

  • 0

    Like

Share Link