Laser Curve Extraction of Wheelset Based on Deep Learning Skeleton Extraction Network

In this paper, a new algorithm for extracting the laser fringe center is proposed. Based on a deep learning skeleton extraction network, the laser stripe center can be extracted quickly and accurately. Skeleton extraction is the process of reducing the shape image to its approximate central axis representation while maintaining the image’s topological and geometric shape. Skeleton extraction is an important step in topological and geometric shape analysis. According to the characteristics of the wheelset laser curve dataset, a new skeleton extraction network, a hierarchical skeleton network (LuoNet), is proposed. The proposed architecture has three levels of the encoder–decoder network, and YE Module interconnection is designed between each level of the encoder and decoder network. In the wheelset laser curve dataset, the F1_score can reach 0.714. Compared with the traditional laser curve center extraction algorithm, the proposed LuoNet algorithm has the advantages of short running time, high accuracy, and stable extraction results.