Journal of Optoelectronics · Laser, Volume. 34, Issue 8, 816(2023)

Research on rail crack detection algorithm based on improved YOLOV4

MIAO Xinfa*, LI Xiaoqin, LIU Baolian, and HOU Yue
Author Affiliations
  • [in Chinese]
  • show less

    Aiming at the small target characteristic of rail surface crack and the low precision and slow speed of traditional detection methods,we propose an object detection method based on improved YOLOV4 network for cracks on the surface of rails in this paper.Firstly,in order to obtain the larger effective receptive field area of the feature map and improve the detection accuracy,we use the improved receptive field block (RFB) module to replace the spatial pyramid pooling (SPP) structure;Secondly,we use the deep separable convolution structure to replace the common convolution structure in the network model,so that the network is lightweight and the detection speed is improved;At the same time,we use K-means + + algorithm to reacquire the anchor frame, and then change the linear scale of the anchor frame to solve the problem that the original anchor frames are not suitable for small target detection.The results show that the mean average precision (mAP) of the improved YOLOV4 is 84.8%,which is 3.4% higher than that of the original YOLOV4 algorithm;The detection speed (FPS) is 62.39 frame/s,which increases by 4.07 frame/s.

    Tools

    Get Citation

    Copy Citation Text

    MIAO Xinfa, LI Xiaoqin, LIU Baolian, HOU Yue. Research on rail crack detection algorithm based on improved YOLOV4[J]. Journal of Optoelectronics · Laser, 2023, 34(8): 816

    Download Citation

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

    Received: May. 30, 2022

    Accepted: --

    Published Online: Sep. 25, 2024

    The Author Email: MIAO Xinfa (1459084294@qq.com)

    DOI:10.16136/j.joel.2023.08.0406

    Topics