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%.
scite is a Brooklyn-based organization that helps researchers better discover and understand research articles through Smart Citations–citations that display the context of the citation and describe whether the article provides supporting or contrasting evidence. scite is used by students and researchers from around the world and is funded in part by the National Science Foundation and the National Institute on Drug Abuse of the National Institutes of Health.