Laser & Optoelectronics Progress, Volume. 60, Issue 12, 1215001(2023)
Yarn-Dyed Fabric Defect Detection Based on U-Shaped Swin Transformer Auto-Encoder
Considering the non-effectiveness of traditional convolution neural networks in detecting pattern defects in yarn-dyed fabrics, a defect detection method based on a U-shaped Swin Transformer reconstruction model and residual analysis is proposed. This method uses the Transformer model to improve the extraction of global image features and enhance reconstruction while solving for the small number and unbalanced types of defective samples during the actual production process. First, the training process of the reconstructed model is completed for a certain pattern using the non-defective samples after adding noise. Subsequently, the test image is inputted into the model to obtain the reconstructed image, and its residual image and reconstructed image are calculated. Finally, the defect areas are detected and located via threshold segmentation and mathematical morphology processing. The results indicate that this method can be effectively used for the detection and location of defect areas on multiple yarn-dyed fabric patterns without requiring the marking of the defective samples.
Get Citation
Copy Citation Text
Yuanyuan Huang, Wenbo Xiong, Hongwei Zhang, Weiwei Zhang. Yarn-Dyed Fabric Defect Detection Based on U-Shaped Swin Transformer Auto-Encoder[J]. Laser & Optoelectronics Progress, 2023, 60(12): 1215001
Category: Machine Vision
Received: Feb. 8, 2022
Accepted: Jun. 13, 2022
Published Online: Jun. 5, 2023
The Author Email: Zhang Hongwei (zhanghongwei@zju.edu.cn)