An algorithm of laser curve segmentation for a train wheelset based on an encoder- decoder network is proposed. Aiming at the rich local features and simple semantic features of the train wheelset laser curve image, a neural network with shallow depth, high resolution, and good detail performance was designed. The proposed neural network makes full use of the dense connection mechanism and the upsampling module to enhance feature reuse and feature propagation. It can extract context semantic information at multiple scales with fewer parameters. Experimental results show that the encoder-decoder network has better performance than other neural networks in laser curve extraction of train wheelset. Based on the encoder-decoder neural network, mIOU, Recall, Accuracy, and F1_score of the laser curve dataset of the train wheelset, the score index reached 86.5%, 89.2%, 99.9%, and 85.0%, which can accurately extract the laser stripe of the train wheelset. Additionally, the encoder-decoder network can diminish the influence of noise on the extraction of laser fringes of a train wheelset to a certain extent. Therefore, it has good application in railway safety.
Read full abstract