Chinese Journal of Liquid Crystals and Displays, Volume. 38, Issue 5, 666(2023)

Rail surface crack detection algorithm based on improved YOLOv5s

Miao-sen ZHOU1,2, Quan-wu TANG1,2、*, Tian-tian SHI1,2, Tong-lan LUO1,2, Ze-xin ZHANG1,2, and Yong-xia XUE1,2
Author Affiliations
  • 1School of Physics and Electronic-Electrical Engineering,Ningxia University,Yinchuan 750021,China
  • 2Ningxia Key Laboratory of Intelligent Sensing for Desert Information,Yinchuan 750021,China
  • show less

    Cracks on the surface of rail sleepers may cause hidden dangers to rail transit. Aiming at the problems of poor universality, low accuracy and low recall of rail crack detection methods, a rail crack detection algorithm YOLOv5s-CBE based on improved YOLOv5s is proposed. Firstly, the CA attention module is added to the backbone C3 module and between C3 and SPPF respectively to capture the channel relationship and location information from the two dimensions of channel and space, so as to improve the feature extraction capability of YOLOv5s backbone network. Secondly, in the neck part of YOLOv5s, BiFPN is used to fuse different scale information to obtain the output feature map with rich semantic information; At the same time, the weighted bi-directional feature fusion pyramid structure adjusts the contribution of input feature maps of different scales to the output by introducing weights, optimizes the feature fusion effect, reduces the loss of feature information in the convolution process, and improves the detection accuracy. Finally, the loss function CIoU in the original yolov5s is changed to EIoU. EIoU not only considers the distance and aspect ratio of the center point, but also considers the real difference in width and height between the prediction frame and the real frame, which improves the prediction accuracy of the anchor frame. Compared with the original network YOLOv5s, the model size of YOLOv5s-CBE rail crack detection network on the self-made rail crack data set is reduced by 1.0 MB, the accuracy mAP is increased by 3.7%, the recall rate is increased from 73.5% to 76.2%, and the phenomenon of missing detection of cracks of different sizes is reduced. It has certain advantages and practical value.

    Tools

    Get Citation

    Copy Citation Text

    Miao-sen ZHOU, Quan-wu TANG, Tian-tian SHI, Tong-lan LUO, Ze-xin ZHANG, Yong-xia XUE. Rail surface crack detection algorithm based on improved YOLOv5s[J]. Chinese Journal of Liquid Crystals and Displays, 2023, 38(5): 666

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Research Articles

    Received: Aug. 13, 2022

    Accepted: --

    Published Online: Jul. 4, 2023

    The Author Email: Quan-wu TANG (tangqw@nxu.edu.cn)

    DOI:10.37188/CJLCD.2022-0267

    Topics