A mixed-attention-based multi-scale autoencoder algorithm for fabric defect detection

Abstract

Aiming at the defects in the process of fabric production, a defect detection model of fabric based on a mixed-attention-based multi-scale non-skipping U-shaped deep convolutional autoencoder (MADCAE) was proposed. In a traditional encoder, the convolutional layer treats each pixel equally, so the importance of different pixels cannot be reflected. It is difficult to obtain richer and more effective information. The reconstruction of the defect region and the detection of the defect region are further affected. In this article, three different scale features of input images are extracted by enlarging the receptive field with large kernel convolution blocks. A hybrid attention module is used to ensure the richness of the extracted information in terms of space and channel. Experiments show that this method can effectively reconstruct fabric parts without requiring a large number of defect marking samples. It can quickly detect and locate defective areas of fabric patterns.