Author:
Feng Xinglong,Gao Xianwen,Luo Ling
Abstract
Abstract
A new Vision Transformer(ViT) model is proposed for the classification of surface defects in hot rolled strip, optimizing the poor learning ability of the original Vision Transformer model on smaller datasets. Firstly, each module of ViT and its characteristics are analyzed; Secondly, inspired by the deep learning model VGGNet, the multilayer fully connected layer in VGGNet is introduced into the ViT model to increase its learning capability; Finally, by performing on the X-SDD hot-rolled steel strip surface defect dataset. The effect of the improved algorithm is verified by comparison experiments on the X-SDD hot-rolled strip steel surface defect dataset. The test results show that the improved algorithm achieves better results than the original model in terms of accuracy, recall, F1 score, etc. Among them, the accuracy of the improved algorithm on the test set is 5.64% higher than ViT-Base and 2.64% higher than ViT-Huge; the accuracy is 4.68% and 1.36% higher than both of them, respectively.
Subject
General Physics and Astronomy
Cited by
8 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献