Research on textile Defect detection Algorithm for Deep Learning

Textile defect detection is a crucial aspect of the textile industry. However, the detection accuracy is often low due to the small size of textile defects and the influence of fabric texture background. To address these issues, we propose a textile defect detection algorithm based on YOLOv8.The reparameterized structure was introduced to the original YOLOv8 to achieve hardware efficiency with the RepGhost module. The head section incorporates a new GD Mechanism and includes the Wasserstein Distance loss function.This method accurately identifies the location and type of defects in textiles with high precision under complex backgrounds, meeting the needs of industrial textile production. We conducted training and comparison experiments on different models using the AITEX public dataset and Roboflow open source dataset. We also designed ablation experiments to verify the considerable improvement of each enhancement on the evaluation index. The experimental results indicate that the detection accuracy of the enhanced YOLOv8 model in the task of detecting textile defects has increased by 18.9\% to 93.8\% on the public dataset. The number of parameters has increased by 10.94MB, and the FLOPs have increased from 8.2G to 11.3G. The improved algorithm has significantly enhanced the accuracy. It can effectively handle large amounts of data and recognize defects even under complex backgrounds, meeting the fast, complex, and high-precision requirements of the textile industry.