1. Deep Learning for the Detection and Recognition of Rail Defects in Ultrasound B-Scan Images
- Author
-
Jidong Yao, Chen Zhengxing, Ping Wang, Yang Kanghua, Liu Yong, Tianle Yu, Qing He, and Wang Qihang
- Subjects
Artificial neural network ,Computer science ,business.industry ,Mechanical Engineering ,Deep learning ,010401 analytical chemistry ,Ultrasound ,02 engineering and technology ,01 natural sciences ,0104 chemical sciences ,0202 electrical engineering, electronic engineering, information engineering ,020201 artificial intelligence & image processing ,Computer vision ,False alarm ,Artificial intelligence ,Detection rate ,business ,Civil and Structural Engineering - Abstract
Rail defect detection is crucial to rail operations safety. Addressing the problem of high false alarm rates and missed detection rates in rail defect detection, this paper proposes a deep learning method using B-scan image recognition of rail defects with an improved YOLO (you only look once) V3 algorithm. Specifically, the developed model can automatically position a box in B-scan images and recognize EFBWs (electric flash butt welds), normal bolt holes, BHBs (bolt hole breaks), and SSCs (shells, spalling, or corrugation). First, the network structure of the YOLO V3 model is modified to enlarge the receptive field of the model, thus improving the detection accuracy of the model for small-scale objects. Second, B-scan image data are analyzed and standardized. Third, the initial training parameters of the improved YOLO V3 model are adjusted. Finally, the experiments are performed on 453 B-scan images as the test data set. Results show that the B-scan image recognition model based on the improved YOLO V3 algorithm reached high performance in its precision. Additionally, the detection accuracy and efficiency are improved compared with the original model and the final mean average precision can reach 87.41%.
- Published
- 2021