Abstract:Defect detection is an essential link in the fabric production process. Due to the diversity of patterns and scarcity of defect samples for colour-patterned fabrics, reconstruction-based unsupervised deep learning algorithms have received extensive attention in the field of fabric defect detection. Among them, unsupervised reconstruction models based on variational autoencoders (VAEs) have been shown to be effective. However, there is a problem of posterior collapse in the process of modelling parametric distributions of continuous variables by VAEs. Therefore, VAE-based defect detection methods for colour-patterned fabrics usually produce ambiguous reconstruction results, thereby affecting the defect detection performance. In this article, an attention-based vector quantisation variational autoencoder (AVQ-VAE) is proposed for colour-patterned fabric defect detection. The method adopts autoregressive modelling of discrete variables to avoid the posterior collapse problem of traditional VAEs, and utilises attention mechanism to enhance the feature representation ability of the model. AVQ-VAE consists of encoder, embedding space, decoder and attention mechanism. The encoder is used to map the input image into multiple feature vectors. Vector quantisation in embedding space is used for discretisation and autoregressive modelling of feature vectors. A decoder is used to decode discrete variables into images of the same size as the original input. Furthermore, an attention mechanism is used to capture channel and spatial correlations, which help the model focus on important information by adaptively recalibrating feature maps. Experimental results on public datasets demonstrate that the proposed method is robust and effective for colour-patterned fabric defect detection.

Fabric Defect Detection Using a One-class Classification Based on Depthwise Separable Convolution Autoencoder

Yarn-dyed Fabric Defect Detection Using U-shaped De-noising Convolutional Auto-Encoder

Colour‐patterned Fabric Defect Detection Based on an Unsupervised Multi‐scale U‐shaped Denoising Convolutional Autoencoder Model

One-Class Model for Fabric Defect Detection

A Machine Vision-based Fabric Defect Detection Solution for Textile Production Industry Using Object Detection

Automatic Fabric Defect Detection with a Multi-Scale Convolutional Denoising Autoencoder Network Model

Visual Saliency Guided Deep Fabric Defect Classification

Unsupervised fabric defect detection based on multiscale image reconstruction and structural similarity assessment

Online Fabric Defects Detection Using Convolutional Neural Networks with Two Frameworks

A Novel Feature‐based Network with Sequential Information for Textile Defect Detection

Color-patterned fabric defect detection algorithm based on triplet attention multi-scale U-shape denoising convolutional auto-encoder

Detection of Microdefects in Fabric with Multifarious Patterns and Colors Using Deep Convolutional Neural Network

Automated fabric defect detection using multi-scale fusion MemAE

Attention-based vector quantisation variational autoencoder for colour-patterned fabrics defect detection

Learning the Color Space for Effective Fabric Defect Detection

Automatic fabric defect detection using a deep convolutional neural network

Fabric Defect Detection Based on Biological Vision Modeling

Unsupervised fabric defect detection based on a deep convolutional generative adversarial network

Colour‐patterned Fabric‐defect Detection Using Unsupervised and Memorial Defect‐free Features

Yarn-dyed fabric defect detection based on an improved autoencoder with Fourier convolution

Semi-supervised fabric defect detection based on image reconstruction and density estimation