Semi-supervised fabric defect detection based on image reconstruction and density estimation
Qihong Zhou, Jun Mei, Qian Zhang, Shaozong Wang, Chen Ge
Abstract
Defective products are a major contributor toward a decline in profits in textile industries. Hence, there are compelling needs for an automated inspection system to identify and locate defects on the fabric surface. Although much effort has been made by researchers worldwide, there are still challenges with computation and accuracy in the location of defects. In this paper, we propose a hybrid semi-supervised method for fabric defect detection based on variational autoencoder (VAE) and Gaussian mixture model (GMM). The VAE model is trained for feature extraction and image reconstruction while the GMM is used to perform density estimation. By synthesizing the detection results from both image content and latent space, the method can construct defect region boundaries more accurately, which are useful in fabric quality evaluation. The proposed method is validated on AITEX and DAGM 2007 public database. Results demonstrate that the method is qualified for automated detection and outperforms other selected methods in terms of overall performance.