Detection of Defects on Metal Surfaces Based on PoolFormer Auto-Encoder

Mengyan Li,Guofeng Qin,Jun Zhu,Yongjian Zhu,Peiwen Mi
DOI: https://doi.org/10.1117/12.3029244
2024-01-01
Abstract:Due to the supervised model data labeling is more complex and cumbersome, this paper adopts unsupervised Auto- Encoder network for defect detection. In addition, for the problem that the convolutional neural network does not have high accuracy in detecting defects on metal surfaces, a defect detection method based on the PoolFormer reconstruction model is proposed. Firstly, according to the type of defects, the defect-free image with superimposed similar morphological noise is used to train the reconstruction model, so that the model has the ability to repair; secondly, the defective image is input into the model for repairing; finally, the residuals between the reconstructed image and the defective image are calculated, and the defect detection and localization can be realized. The experimental results show that the method is able to detect many types of metal surface defects, and the detection accuracy reach more than 98%.
What problem does this paper attempt to address?