Paper Info
Title
Real-Time Implementation Of Fabric Defect Detection Based On Variational Automatic Encoder With Structure Similarity
Abstract
Automatic detection of fabric defects based on machine vision is an important topic in the quality control of cotton textile factories. There are many kinds of defects in fabric production, it is very difficult to classify the defects automatically. In recent years, deep learning image processing technology based on a convolutional neural network (CNN) can train and extract features of the target image automatically. Since a large number of defect samples cannot be collected completely, we compared unsupervised learning algorithms based on CNN, including auto encoder (AE), variational automatic encoder (VAE), and generative adversarial networks (GAN). Because of the large amount of calculation and the difficulty of training in GAN, we chose AE and VAE codec networks and then introduced mean structural similarity (MSSIM) as network training loss function to improve the performance that only used L-p-distance loss function for image brightness comparison. After training finished, the authors used the trained model to obtain target defects from SSIM residual maps between input and reconstruct images. According to the evaluation results, we finally implemented a fabric defect detection system based on VAE on Jetson TX2 from Nvidia Corporation, USA. The optimized algorithm can meet the real-time requirements of the project and realize its popularization and application.
Year
DOI
Venue
2021
10.1007/s11554-020-01023-5
JOURNAL OF REAL-TIME IMAGE PROCESSING
Keywords
DocType
Volume
Fabric defects, Variational automatic encoder, Structural similarity, Real-time, Deep learning
Journal
18
Issue
ISSN
Citations 
3
1861-8200
0
PageRank 
References 
Authors
0.34
0
4
Name
Order
Citations
PageRank
Wei Wei124.07
Dexiang Deng2694.43
Lin Zeng300.34
Chen Zhang4112.13