Lightweight Fabric Defect Detection Based on Improved YOLOv4
Authors: 
Qing Ni, Xin Luo, Ran Tao, Youqun ShiAuthors Info & Claims
Qing Ni, Xin Luo, Ran Tao, Youqun ShiAuthors Info & ClaimsPages 288 - 294
Abstract
Fabric quality is crucial for subsequent production, so automated detection of different types of defects on fabric surfaces is essential. To improve the applicability of the detection model in textile factories, a lightweight fabric defect detection method based on YOLOv4 is proposed, which firstly addresses the characteristics of large scale transformation of fabric defects and more small defects by combining the shallow features of the 11th layer in YOLOv4 with the deep features to generate a new scale feature layer; secondly, the SPP structure in YOLOv4 is improved to using pooling decomposition to improve the large pooling kernel and increase the model computation speed; finally, the depth-separable convolution is introduced to optimize the model size, making the model detection faster to meet the actual needs of textile mills. The improved network improves the defect detection accuracy, with an average detection accuracy of 79.31%, and its detection speed is also 5 times faster than that of the original YOLOv4, which is more suitable for the actual detection of textile mills than the original YOLOv4.
References
[1]
Bochkovskiy, Alexey, Chien-Yao Wang, and Hong-Yuan Mark Liao. 2020. Yolov4: Optimal Speed and Accuracy of Object Detection. CoRR abs/2004.10934 (/ 2020). https://arxiv.org/abs/2004.10934.
[2]
Duan, Kaiwen, Song Bai, Lingxi Xie, Honggang Qi, Qingming Huang, and Qi Tian. 2019. Centernet: Keypoint Triplets for Object Detection. 2019, https://doi.org/10.1109/ICCV.2019.00667.
[3]
Girshick, R. 2015. Fast R-Cnn. International Conference on Computer Vision.
[4]
Girshick, R., J. Donahue, T. Darrell, and J. Malik. 2013. Rich Feature Hierarchies for Accurate Object Detection and Semantic Segmentation. IEEE Computer Society.
[5]
Han, Wan-Jiang, Li-Xin Jiang, Tian-Bo Lu, Xiaoyan Zhang, and Sun Yi. 2014. Study on Residual Defect Prediction Using Multiple Technologies. Journal of Advances in Information Technology 5 ( August, 2014): 79-85. https://dx.doi.org/
[6]
He, K., X. Zhang, S. Ren, and J. Sun. 2015. Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition. IEEE Transactions on Pattern Analysis and Machine Intelligence 37, no. 9 (September 2015): 1904-16. https://dx.doi.org/10.1109/TPAMI.2015.2389824.
[7]
Howard, Andrew G., Menglong Zhu, Bo Chen, Dmitry Kalenichenko, Weijun Wang, Tobias Weyand, Marco Andreetto, and Hartwig Adam. 2017. Mobilenets: Efficient Convolutional Neural Networks for Mobile Vision Applications. CoRR abs/1704.04861 (2017). http://arxiv.org/abs/1704.04861.
[8]
Jiang, Ban, Chunlei Li, Zhoufeng Liu, Aihua Zhang, and Yan Yang. 2020. Fabric Defect Detection Based on Total Variation Regularized Double Low-Rank Matrix Representation. Bio-inspired Computing: Theories and Applications. Edited by Linqiang Pan, Jing Liang, and Boyang Qu. Singapore: Springer Singapore.
[9]
Liang, Ying, Ke Xu, and Peng Zhou.2020. Mask Gradient Response-Based Threshold Segmentation for Surface Defect Detection of Milled Aluminum Ingot. Sensors 20 (August 2020): 4519. https://dx.doi.org/10.3390/s20164519.
[10]
Lin, Zhongkang, Zhiqiang Guo, and Jie Yang. 2019. Research on Texture Defect Detection Based on Faster-Rcnn and Feature Fusion. Icmlc '19 (2019): 429–33. https://dx.doi.org/10.1145/3318299.3318341.
[11]
Liu, Qiang, Chuan Wang, Yusheng Li, Mingwang Gao, and Jingao Li. 2022. A Fabric Defect Detection Method Based on Deep Learning. IEEE Access 10 ( 2022): 4284-96. https://dx.doi.org/10.1109/ACCESS.2021.3140118.
[12]
Liu, Wei, Dragomir Anguelov, Dumitru Erhan, Christian Szegedy, Scott E. Reed, Cheng-Yang Fu, and Alexander C. Berg. 2016. Ssd: Single Shot Multibox Detector. 2016, https://doi.org/10.1007/978-3-319-46448-0_2.
[13]
Redmon, Joseph, Santosh Kumar Divvala, Ross B. Girshick, and Ali Farhadi. 2016. You Only Look Once: Unified, Real-Time Object Detection. 2016, https://doi.org/10.1109/CVPR.2016.91.
[14]
Redmon, Joseph, and Ali Farhadi. 2017. Yolo9000: Better, Faster, Stronger. https://doi.org/10.1109/CVPR.2017.690
[15]
Redmon, Joseph, and Ali Farhadi. 2018. Yolov3: An Incremental Improvement. CoRR abs/1804.02767 (2018). http://arxiv.org/abs/1804.02767.
[16]
Ren, S., K. He, R. Girshick, and J. Sun. 2016. Faster R-Cnn: Towards Real-Time Object Detection with Region Proposal Networks. Vol. 28. NIPS.
[17]
Yao, Hongge, Qin Na, Shuangwu Zhu, Min Lin, and Jun Yu. 2021. Fabric Defect Detection with an Attention Mechanism Based on Hard Sample Training. Textile Research Journal (2021). https://dx.doi.org/
Index Terms
- Lightweight Fabric Defect Detection Based on Improved YOLOv4
Recommendations
Fabric Defect Detection Based on Cascade Faster R-CNN
CSAE '20: Proceedings of the 4th International Conference on Computer Science and Application EngineeringWith the development of the textile production industry, quality inspection has become an increasingly important means of ensuring the quality of textiles. In order to solve the problem of low efficiency of traditional manual detection methods, ...
Fabric Defect Target Detection Algorithm Based on YOLOv4 Improvement
Web Information Systems and ApplicationsAbstractFabric 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 ...
Fabric Defect Detection Based on Lightweight Neural Network
Pattern Recognition and Computer VisionAbstractOwing 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 ...
Comments
Information & Contributors
Information
Published In
March 2023
824 pages
ISBN:9781450399029
DOI:10.1145/3594315
Copyright © 2023 Owner/Author.
Permission to make digital or hard copies of part or all of this work for personal or classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation on the first page. Copyrights for third-party components of this work must be honored. For all other uses, contact the Owner/Author.
Publisher
Association for Computing Machinery
New York, NY, United States
Publication History
Published: 02 August 2023
Check for updates
Qualifiers
- Research-article
- Research
- Refereed limited
Funding Sources
Conference
ICCAI 2023
ICCAI 2023: 2023 9th International Conference on Computing and Artificial Intelligence
March 17 - 20, 2023
Tianjin, China
Contributors
Other Metrics
Bibliometrics & Citations
Bibliometrics
Article Metrics
- 0Total Citations
- 35Total Downloads
- Downloads (Last 12 months)30
- Downloads (Last 6 weeks)4
Reflects downloads up to 04 Oct 2024
Other Metrics
Citations
View Options
Get Access
Login options
Check if you have access through your login credentials or your institution to get full access on this article.
Sign inFull Access
View options
View or Download as a PDF file.
PDFeReader
View online with eReader.
eReaderHTML Format
View this article in HTML Format.
HTML Format