Online Rail Fastener Detection Based on YOLO Network  

作　　者：Jun Li Xinyi Qiu Yifei Wei Mei Song Xiaojun Wang 

机构地区：[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

出　　处：《Computers, Materials & Continua》2022年第9期5955-5967,共13页计算机、材料和连续体（英文）

基　　金：This work was supported by the National Natural Science Foundation of China(61871046,SM,http://www.nsfc.gov.cn/).

摘　　要：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.

关 键 词：Fastener detection deep learning state recognition real-time classification 

分 类 号：TP393[自动化与计算机技术—计算机应用技术]