Modern Manufacturing Engineering ›› 2024, Vol. 528 ›› Issue (9): 127-135.doi: 10.16731/j.cnki.1671-3133.2024.09.017

Previous Articles     Next Articles

Research on image segmentation algorithm of high-speed rail carbon skateboard based on machine vision

LIU Weimin1, ZHANG Shaoning1, ZHENG Aiyun1, LIU Jin2, ZHENG Zhi1   

  1. 1 College of Mechanical Engineering,North China University of Science and Technology, Tangshan 063210,China;
    2 CRRC Tangshan Co.,Ltd.,Tangshan 064000,China
  • Received:2024-01-03 Online:2024-09-18 Published:2024-09-27

Abstract: A novel Swin Transformer semantic segmentation optimization algorithm with a codec structure is proposed to solve the problems such as difficulty in identifying the edge of carbon sliding plate by semantic segmentation model,large interference in complex background,and serious feature information loss. Firstly,the backbone network adopts U-shaped codec structure to realize multi-scale information fusion.Secondly,the attention local enhancement module is added to expand the sensing field and improve the model generalization ability. Then,the upsampling structure with data correlation is used to enhance the quality of upsampling,eliminate the impact of resolution on prediction results,and improve image reconstruction capability. Finally,the skip connection is replaced by the residual path to make the semantic information in the codec structure more closely connected and improve the training efficiency. The experimental results show that the Swin Transformer semantic segmentation algorithm improves the measurement prediction accuracy by 3.63 %,and the average accuracy of pixel classification in all categories is improved by 7.29 %. The research results confirm the superiority and robustness of the Swin Transformer semantic segmentation model in identifying and handling carbon slide tasks.

Key words: carbon sliding plate, Swin Transformer, local enhanced sensing, image reconstruction, semantic segmentation

CLC Number: 

Copyright © Modern Manufacturing Engineering, All Rights Reserved.
Tel: 010-67126028 E-mail: 2645173083@qq.com
Powered by Beijing Magtech Co. Ltd