• Laser & Optoelectronics Progress
  • Vol. 58, Issue 4, 0410020 (2021)
Li Zhang1, Danping Huang1、*, Shipeng Liao2, Shaodong Yu1、3, Jianqiu Ye1, Xin Wang1, and Na Dong1
Author Affiliations
  • 1School of Mechanical Engineering, Sichuan University of Science & Engineering, Yibin, Sichuan 644000, China;
  • 2Chengdu Institute of Computer Application, Chinese Academy of Sciences, Chengdu, Sichuan 610041, China
  • 3School of Mechanical Engineering, Sichuan University, Chengdu, Sichuan 610065, China
  • show less
    DOI: 10.3788/LOP202158.0410020 Cite this Article Set citation alerts
    Li Zhang, Danping Huang, Shipeng Liao, Shaodong Yu, Jianqiu Ye, Xin Wang, Na Dong. Wheelset Tread Defect Detection Method Based on Target Detection Network[J]. Laser & Optoelectronics Progress, 2021, 58(4): 0410020 Copy Citation Text show less

    Abstract

    It is difficult to quickly and accurately identify wheelset tread defects using traditional image processing algorithms. We propose an algorithm to accomplish this using a dual deep neural network. The dual network is divided into a tread-extraction network and a defect-identification network. Based on the characteristics of the treads as a big target, we analyze and test the SSD network, and apply this network to extract the tread area from wheelset images. To improve the efficiency of tread defect recognition, after the tread image is extracted, we optimize the YOLOv3 network structure to obtain M-YOLOv3 for the characteristics of medium and small tread defect targets. The experimental results show that when extracting tread areas, the average precision (AP) of the SSD algorithm is the highest (99.8%). When identifying tread defects, the AP of the M-YOLOv3 network reaches 89.9%. Compared with the original YOLOv3, the image computing time of the M-YOLOv3 network is reduced by 7.1%, with the AP showing only a 0.6% loss. The results demonstrate the proposed algorithm’s high detection accuracy.
    Li Zhang, Danping Huang, Shipeng Liao, Shaodong Yu, Jianqiu Ye, Xin Wang, Na Dong. Wheelset Tread Defect Detection Method Based on Target Detection Network[J]. Laser & Optoelectronics Progress, 2021, 58(4): 0410020
    Download Citation