U-shaped Vision Transformer and Its Application in Gear Pitting Measurement

查看全文 作　　者：Sijun [1]Wang;Yi [1]Qin;Dejun [1]Xi;Chen [1]Liang

机构地区：[1]College of Mechanical and Vehicle Engineering,Chongqing University,Chongqing,China

出　　处：《Journal of Dynamics, Monitoring and Diagnostics》索引2022年第1卷第4期,共6页

基　　金：supported in part by National Natural Science Foundation of China under Grants 62033001 and 52175075. 摘　　要：Although convolutional neural networks have become the mainstream segmentation model,the locality of convolution makes them cannot well learn global and long-range semantic information.To further improve the performance of segmentation models,we propose U-shaped vision Transformer(UsViT),a model based on Transformer and convolution.Specifically,residual Transformer blocks are designed in the encoder of UsViT,which take advantages of residual network and Transformer backbone at the same time.What is more,transpositions in each Transformer layer achieve the information interaction between spatial locations and feature channels,enhancing the capability of feature learning.In the decoder,for enhancing receptive field,different dilation rates are introduced to each convolutional layer.In addition,residual connections are applied to make the information propagation smoother when training the model.We first verify the superiority of UsViT on automatic portrait matting public dataset,which achieves 90.43%accuracy(Acc),95.56%Dice similarity coefficient,and 94.66%Intersection over Union with relatively fewer parameters.Finally,UsViT is applied to gear pitting measurement in gear contact fatigue test,and the comparative results indicate that UsViT can improve the Acc of pitting detection. 关键词：vision Transformer residual connection dilation rate information interaction pitting measurement

相关文献

参考文献（28）
引证文献（1）
耦合文献（0）

参考文献(28)

引证文献(1)

互联网出版许可证新出网证(渝)字10号全国400电话 - 免长途话费

U-shaped Vision Transformer and Its Application in Gear Pitting Measurement

参考文献(28)

引证文献(1)

用户提示