Abstract
Fabric defect detection is a key part of product quality assessment in the textile industry. It is important to achieve fast, accurate and efficient detection of fabric defects to improve productivity in the textile industry. For the problems of varying scales, irregular shapes and many small objects, an improved YOLOv4 object detection algorithm for fabric defects is proposed. Firstly, in order to improve the detection accuracy of small objects, the RFB module is introduced and fused with shallow features, which can obtain receptive fields of different scales to improve the features extracted from the backbone network. Secondly, the introduction of spatial and channel attention mechanisms can enhance fused features, allowing the network to focus on useful information. Experimental results show that the mean average precision of the improved YOLOv4 object detection algorithm in fabric defect map detection is 71.89%. The improved algorithm can accurately improve the accuracy of fabric defect positioning.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Zhang, H., Wang, K.F., Wang, F.Y.: Advances and perspectives on applications of deep learning in visual object detection. Zidonghua Xuebao/Acta Automatica Sinica 43(8), 1289–1305 (2017)
Chen, R., Jin, Yu., Xu, L.: A classroom student counting system based on improved context-based face detector. In: Wang, G., Lin, X., Hendler, J., Song, W., Xu, Z., Liu, G. (eds.) WISA 2020. LNCS, vol. 12432, pp. 326–332. Springer, Cham (2020). https://doi.org/10.1007/978-3-030-60029-7_30
Bochkovskiy, A., Wang, C.Y., Liao, H.Y.M.: YOLOv4: optimal speed and accuracy of object detection. arXiv:2004.10934 (2020)
Yapi, D., Allili, M.S., et al.: Automatic fabric defect detection using learning-based local textural distributions in the contourlet domain. IEEE Trans. Autom. Sci. Eng. 15(3), 1014–1026 (2017)
Vaibhav, M., Karlekar, V., Bhangale, K., et al.: Fabric defect detection using wavelet filter. In: 2015 International Conference on Computing Communication Control and Automation. IEEE (2015)
Jia, L., Chen, C., Liang, J., et al.: Fabric defect inspection based on lattice segmentation and Gabor filtering. Neurocomputing 238(MAY17), 84–102 (2017)
Deng, C., Liu, Y.: Defect detection of twill cloth based on edge detection. Meas. Control Technol. 37(12), 110–113 (2018)
Girshick, R., Donahue, J., Darrell, T., Malik, J.: 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, June 2014
He, K., Zhang, X., Ren, S., et al.: Spatial pyramid pooling in deep convolutional networks for visual recognition. IEEE Trans. Pattern Anal. Mach. Intell. 37(9), 1904–16 (2014)
Girshick, R.: Fast R-CNN. In: Proceedings of the IEEE International Conference on Computer Vision (ICCV), pp. 1440–1448, December 2015
Ren, S., He, K., Girshick, R., et al.: Faster R-CNN: towards real-time object detection with region proposal networks. In: Advances in Neural Information Processing Systems, pp. 91–99 (2015)
Redmon, J., Divvala, S., Girshick, R., et al.: You only look once: unified, real-time object detection. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 779–788 (2015)
Liu, W., et al.: SSD: single shot MultiBox detector. In: Leibe, B., Matas, J., Sebe, N., Welling, M. (eds.) ECCV 2016. LNCS, vol. 9905, pp. 21–37. Springer, Cham (2016). https://doi.org/10.1007/978-3-319-46448-0_2
Redmon, J., Farhadi, A.: YOLO9000: better, faster, stronger. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 7263–7271 (2017)
Redmon, J., Farhadi, A.: YOLOv3: an incremental improvement. arXiv preprint arXiv:1804.02767 (2018)
Lin, T.Y., Goyal, P., Girshick, R., et al.: Focal loss for dense object detection. IEEE Trans. Pattern Anal. Mach. Intell. (99), 2999–3007 (2017)
Woo, S., Park, J., Lee, J., Kweon, I.S.: CBAM: convolutional block attention module. In: Proceedings of the European Conference on Computer Vision (ECCV), pp. 3–19 (2018)
Jie, H., Li, S., Gang, S.: Squeeze-and-excitation networks. In: 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR). IEEE (2018)
Neubeck, A., Gool, L.: Efficient non-maximum suppression. In: International Conference on Pattern Recognition (2006)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2021 Springer Nature Switzerland AG
About this paper
Cite this paper
Wang, Y., Hao, Z., Zuo, F., Su, Z. (2021). Fabric Defect Target Detection Algorithm Based on YOLOv4 Improvement. In: Xing, C., Fu, X., Zhang, Y., Zhang, G., Borjigin, C. (eds) Web Information Systems and Applications. WISA 2021. Lecture Notes in Computer Science(), vol 12999. Springer, Cham. https://doi.org/10.1007/978-3-030-87571-8_56
Download citation
DOI: https://doi.org/10.1007/978-3-030-87571-8_56
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-87570-1
Online ISBN: 978-3-030-87571-8
eBook Packages: Computer ScienceComputer Science (R0)