Affiliation:
1. Techanical and Electrical Engineering Department, Shanxi Institute of Energy, Taiyuan 030600, China
2. College of Mechanical Engineering, North University of China, Taiyuan 030000, China
Abstract
Aiming at the problems of low-resolution steel surface defects imaging, such as defect type confusion, feature blurring, and low classification accuracy, this paper proposes an autocorrelation semantic enhancement network (ASENet) for the classification of steel surface defects. It mainly consists of a backbone network and an autocorrelation semantic enhancement module (ASE), in which the autocorrelation semantic enhancement module consists of three main learnable modules: the CS attention module, the autocorrelation computation module, and the contextual feature awareness module. Specifically, we first use the backbone network to extract the basic features of the image and then use the designed CS attention module to enhance the basic features. In addition, to capture different aspects of semantic objects, we use the autocorrelation module to compute the correlation between neighborhoods and contextualize the basic and augmented features to enhance the recognizability of the features. Experimental results show that our method produces significant results, and the classification accuracy reaches 96.24% on the NEU-CLS-64 dataset. Compared with ViT-B/16, Swin_t, ResNet50, Mobilenet_v3_small, Densenet121, Efficientnet_b2, and baseline, the accuracy is 9.43%, 5.15%, 4.87%, 3.34%, 3.28%, 3.01%, and 2.72% higher, respectively.
Subject
Materials Chemistry,Surfaces, Coatings and Films,Surfaces and Interfaces
Reference48 articles.
1. The use of stainless steel in structures;Gardner;Prog. Struct. Eng. Mater.,2010
2. Wen, X., Shan, J., He, Y., and Song, K. (2022). Steel surface defect recognition: A survey. Coatings, 13.
3. Multi-class classification for steel surface defects based on machine learning with quantile hyper-spheres;Chu;Chemom. Intell. Lab. Syst.,2017
4. Machine learning-based imaging system for surface defect inspection;Park;Int. J. Precis. Eng.-Manuf.-Green Technol.,2016
5. Review of surface defect detection of steel products based on machine vision;Tang;IET Image Process.,2023