Research on improving YOLOv5s algorithm for fabric defect detection

Abstract

PurposeIn the fabric manufacturing industry, various unfavorable factors, including machine fault and yarn breakage, can easily cause fabric defects and affect product quality, begetting huge economic losses to enterprises. Thus, automatic fabric defect detection systems have become an important development direction. Herein, the most common defects in the fabric production process, like ribbon yarn, broken yarn, cotton ball, holes, yarn shedding and stains, are detected. Current fabric defect detection systems afford low detection accuracy and a high missed detection rate for small target fabric defects. Therefore, this study proposes deep learning technology for automatically detecting fabric defects by improving the YOLOv5s target detection algorithm. The improved algorithm is termed YOLOv5s-4SCK, which can effectively detect fabric defects. This study aims to discuss the aforementioned issues.Design/methodology/approachSpecifically, based on the YOLOv5s algorithm, first, the structure of YOLOv5s is modified to add a small target detection layer, fully utilize deep and shallow features and reduce the missed detection rate of small target fabric defects. Second, the integration of CARAFE upsampling enables the effective retention of feature information and maintenance of a certain computational efficiency, thereby improving the detection accuracy. Finally, the K-Means++ clustering algorithm is used to analyze the position of the center point of the prior box to better obtain the anchor box and improve the average accuracy and evaluation index of detection.FindingsThe research results show that the YOLOv5s-4SCK algorithm increases the accuracy by 4.1% and the detection speed by 2 f.s-1 compared to the original YOLOv5s algorithm, and it effectively improves the original YOLOv5s problem of high missed detection rate of small targets.Research limitations/implicationsThe YOLOv5s-4SCK proposed in this paper can effectively reduce the missed detection rate of fabric defects, improve the detection efficiency and has certain industrial value.Practical implicationsThe proposed algorithm can quickly identify fabric defects, effectively improving the detection rate. In the future, the proposed algorithm will be applied in the actual industry.Social implicationsAutomatic fabric defect detection reduces the manpower of inspectors, and the proposed YOLOv5s-4SCK algorithm is also suitable for other recognition fields.Originality/valueThe proposed YOLOv5s-4SCK algorithm has been tested using real cloth to ensure its accuracy, and its performance is better than the original YOLOv5s algorithm.