Abstract
Owing to the variety and complexity of defects in the fabric texture image, automatic fabric defect detection is a challenging task in the fields of machine vision. Deep convolutional neural network (CNN) has achieved remarkable progress in the field of target detection, and the application of deep CNN model to fabric defect detection has achieved better results. However, with the detection accuracy increasing of deep CNNs, comes the drawbacks of significant increase in computational costs and storage services, which seriously hinders the usages of CNN on resource-limited environments such as mobile or embedded devices. In this paper, a fabric defect detection method using lightweight CNN is proposed. We introduce an extremely computation-efficient CNN architecture named YOLO-LFD, which adopts continuous 3 \(\times \) 3 and 1 \(\times \) 1 convolution layers for dimensionality reduction and feature fusion. We use multi-scale feature extraction method to improve the detection ability of the model for different size targets, and adopt K-means clustering method on the fabric defect image dataset to find the optimal size of anchor boxes. Experimental results demonstrate that the proposed scheme improves the detection accuracy of fabric defects, greatly reduces the model parameters, and improves the detection speed, which can provide real-time fabric defects detection on embedded devices.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Li, Y., Zhao, W., Pan, J.: Deformable patterned fabric defect detection with fisher criterion-based deep learning. IEEE Trans. Autom. Sci. Eng. 14(2), 1256–1264 (2017)
Zhao, X., Deng, Z., Ran, Y.: Detection of fabric defect based on weighted local entropy. Cotton Text. Technol. 40(10), 20–23 (2012)
Xiaobo, Y.: Fabric defect detection of statistic aberration feature based on GMRF model. J. Text. Res. 34(4), 137–142 (2013)
Shuai, Z., Fengsheng, Z., Fucai, L.: Feature extraction of fabric defects based on wavelet transform. J. QingDao Univ. 28(2), 53–59 (2013)
Yuan, L., Li, F., Yang, Y., et al.: Analysis of texture feature extracted by gray level co-occurrence matrix. Comput. Appl. 29(4), 1019–1021 (2009)
Lecun, Y., Bengio, Y., Hinton, G.: Deep learning. Nature 521(7553), 436 (2015)
Jing, J., Dong, A., Li, P., et al.: Yarn-dyed fabric defect classification based on convolutional neural network. Opt. Eng. 56(9), 093104 (2017)
Shi, Y., Li, Y., Wei, X., et al.: A faster-RCNN based chemical fiber paper tube defect detection method. In: 2017 5th International Conference on Enterprise Systems (ES), pp. 173–177. IEEE (2017)
Tao, X., Zhang, D., Ma, W., et al.: Automatic metallic surface defect detection and recognition with convolutional neural networks. Appl. Sci. 8(9), 1575 (2018)
Girshick, R., Donahue, J., Darrell, T., et al.: Rich feature hierarchies for accurate object detection and semantic segmentation. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 580–587 (2014)
Girshick, R.: Fast R-CNN. Comput. Sci. (2015)
Lin, T.Y., Dollr, P., Girshick, R., et al.: Feature pyramid networks for object detection (2016)
Han, S., Liu, X., Mao, H., et al.: EIE: efficient inference engine on compressed deep neural network. In: ISCA. IEEE Computer Society (2016)
He, K., Zhang, X., Ren, S., et al.: Deep residual learning for image recognition (2015)
Redmon, J., Farhadi, A.: YOLO9000: better, faster, stronger. In: 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 6517–6525. IEEE (2017)
Szegedy, C., Liu, W., Jia, Y., et al.: Going deeper with convolutions. In: 2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR). IEEE (2015)
Dhanachandra, N., Manglem, K., Chanu, Y.J.: Image segmentation using K-means clustering algorithm and subtractive clustering algorithm. Proc. Comput. Sci. 54, 764–771 (2015)
Nowozin, S.: Optimal decisions from probabilistic models: the intersection-over-union case. In: IEEE Conference on Computer Vision and Pattern Recognition. IEEE Computer Society (2014)
Ahmed, F., Tarlow, D., Batra, D.: Optimizing expected intersection-over-union with candidate-constrained CRFs. In: IEEE International Conference on Computer Vision (2016)
Lin, T.Y., et al.: Microsoft COCO: common objects in context. In: Fleet, D., Pajdla, T., Schiele, B., Tuytelaars, T. (eds.) ECCV 2014. LNCS, vol. 8693, pp. 740–755. Springer, Cham (2014). https://doi.org/10.1007/978-3-319-10602-1_48
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2019 Springer Nature Switzerland AG
About this paper
Cite this paper
Liu, Z., Cui, J., Li, C., Wei, M., Yang, Y. (2019). Fabric Defect Detection Based on Lightweight Neural Network. In: Lin, Z., et al. Pattern Recognition and Computer Vision. PRCV 2019. Lecture Notes in Computer Science(), vol 11857. Springer, Cham. https://doi.org/10.1007/978-3-030-31654-9_45
Download citation
DOI: https://doi.org/10.1007/978-3-030-31654-9_45
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-31653-2
Online ISBN: 978-3-030-31654-9
eBook Packages: Computer ScienceComputer Science (R0)