Journal of Optoelectronics · Laser, Volume. 36, Issue 7, 733(2025)

Lightweight train wheelset tread defect detection method based on improved YOLOv7-tiny

QIAN Shangle1,2, CAO Wei3, and GAO Junwei1,2、*
Author Affiliations
  • 1School of Automation, Qingdao University, Qingdao, Shandong 266071, China
  • 2Shandong Key Laboratory of Industrial Contral Technology, Qingdao, Shandong 266071, China
  • 3Qingdao International Airport Group Co Ltd., Qingdao, Shandong 266300, China
  • show less

    Aiming at the problems of low detection accuracy, slow detection rate and single detection category of wheelset tread defects, an improved lightweight YOLOv7-tiny tread damage detection method is proposed. In this method, the lightweight MobileNetV3 network is used to replace the backbone network of YOLOv7-tiny, and the parameter number and calculation amount of the model are reduced. Embedding BiFormer attention mechanism into the backbone network can strengthen the features of the learning target region and improve the detection accuracy of the model. The centralized feature pyramid (CFP) is used to enhance the feature′s in-layer adjustment ability, capture the global long distance dependence and local critical information of tread defects. Wise intersection over union (WIoU) loss function is employed to accelerate the convergence rate of border regression loss and enhance the robustness of the model. GSconv decoupled head (GSDH) is introduced into the YOLOv7-tiny detection header to decouple separated feature channels from classification and regression tasks, effectively improving the parallel computation rate and detection accuracy of the model. The experimental results show that the improved YOLOv7-tiny network parameter number and computation amount are reduced by 11.7% and 21.2% respectively, the detection precision is increased by 7.9%, the recall rate is increased by 10.5%, the mean average precision is increased by 10.1%, and the frame per second is increased by 7.4 frames/s, which realizes lightweight and has better detection performance. The improved method has a wide application prospect in wheelset tread damage detection.

    Tools

    Get Citation

    Copy Citation Text

    QIAN Shangle, CAO Wei, GAO Junwei. Lightweight train wheelset tread defect detection method based on improved YOLOv7-tiny[J]. Journal of Optoelectronics · Laser, 2025, 36(7): 733

    Download Citation

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

    Category:

    Received: Apr. 3, 2024

    Accepted: Jun. 24, 2025

    Published Online: Jun. 24, 2025

    The Author Email: GAO Junwei (qdgao163@163.com)

    DOI:10.16136/j.joel.2025.07.0163

    Topics