6 September 2023 Transformer-based visual inspection algorithm for surface defects
Qinmiao Zhu, Jingyun Chang, Teng Liu, Yuhui Wang, Hua Yang
Author Affiliations +
Abstract

Industrial production often faces a variety of complex working conditions that lead to various defects, including Mura, on the surfaces of various industrial products. We propose a reconstruction network called RecTransformer, which is developed with a transformer for anomaly inpainting. RecTransformer is designed to effectively detect various types of surface defects despite using only a small number of defect samples. RecTransformer simplifies the defect detection problem to a patch-level image completion problem. Without using convolution, the given block image is processed by the transformer model to generate a defect-free reconstructed image. Herein, global semantic information is established, and an attention mechanism is built in the patch sequence, and the spatial information of the patches is determined by position encoding to complete the global image reconstruction process. With a limited number of defect samples as training data, the RecTransformer algorithm accurately reconstructs defects. It achieves an area under the receiver operating characteristic curve score of 97.6% for pixel-level segmentation on the testing dataset. Experiments conducted on a universal surface defect dataset demonstrate the effectiveness of the RecTransformer algorithm. RecTransformer can be adapted to detect various types of surface defects, including Mura in display devices, with only a small number of defect samples.

© 2023 Society of Photo-Optical Instrumentation Engineers (SPIE)
Qinmiao Zhu, Jingyun Chang, Teng Liu, Yuhui Wang, and Hua Yang "Transformer-based visual inspection algorithm for surface defects," Optical Engineering 62(9), 094102 (6 September 2023). https://doi.org/10.1117/1.OE.62.9.094102
Received: 27 March 2023; Accepted: 12 June 2023; Published: 6 September 2023
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Transformers

Image restoration

Image processing

Education and training

Defect detection

Reconstruction algorithms

Visual inspection

RELATED CONTENT


Back to Top