期刊
JOURNAL OF REAL-TIME IMAGE PROCESSING
卷 18, 期 3, 页码 807-823出版社
SPRINGER HEIDELBERG
DOI: 10.1007/s11554-020-01023-5
关键词
Fabric defects; Variational automatic encoder; Structural similarity; Real-time; Deep learning
This paper introduces the importance of automatic detection of fabric defects based on machine vision and the use of deep learning image processing technology to train and extract features of target images. By comparing different unsupervised learning algorithms, AE and VAE were chosen to implement the algorithm, and a fabric defect detection system based on VAE was finally implemented on NVIDIA's Jetson TX2.
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.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据