26 September 2017 Yarn-dyed fabric defect classification based on convolutional neural network
Author Affiliations +
Abstract
Considering that manual inspection of the yarn-dyed fabric can be time consuming and inefficient, we propose a yarn-dyed fabric defect classification method by using a convolutional neural network (CNN) based on a modified AlexNet. CNN shows powerful ability in performing feature extraction and fusion by simulating the learning mechanism of human brain. The local response normalization layers in AlexNet are replaced by the batch normalization layers, which can enhance both the computational efficiency and classification accuracy. In the training process of the network, the characteristics of the defect are extracted step by step and the essential features of the image can be obtained from the fusion of the edge details with several convolution operations. Then the max-pooling layers, the dropout layers, and the fully connected layers are employed in the classification model to reduce the computation cost and extract more precise features of the defective fabric. Finally, the results of the defect classification are predicted by the softmax function. The experimental results show promising performance with an acceptable average classification rate and strong robustness on yarn-dyed fabric defect classification.
© 2017 Society of Photo-Optical Instrumentation Engineers (SPIE)
Junfeng Jing, Amei Dong, Pengfei Li, Kaibing Zhang, "Yarn-dyed fabric defect classification based on convolutional neural network," Optical Engineering 56(9), 093104 (26 September 2017). https://doi.org/10.1117/1.OE.56.9.093104 . Submission: Received: 2 July 2017; Accepted: 29 August 2017
Received: 2 July 2017; Accepted: 29 August 2017; Published: 26 September 2017
JOURNAL ARTICLE
9 PAGES


SHARE
Back to Top