Abstract:This article aims to improve deep-learning-based surface defect recognition. Owing to the insufficiency of the defect images in practical production lines and the high cost of labeling, it is difficult to obtain a sufficient defect data set in terms of diversity and quantity. A new generation method called surface defect-generation adversarial network (SDGAN), which employs generative adversarial networks (GANs), is proposed to generate defect images using a large number of defect-free images from industrial sites. Experiments show that the defect images generated by the SDGAN have better image quality and diversity than those generated by the state-of-the-art methods. The SDGAN is applied to expand the commutator cylinder surface defect image data sets with and without labels (referred to as the CCSD-L and CCSD-NL data sets, respectively). Regarding anomaly recognition, a 1.77% error rate and a 49.43% relative improvement (IMP) for the CCSD-NL defect data set are obtained. Regarding defect classification, a 0.74% error rate and a 57.47% IMP for the CCSD-L defect data set are achieved. Moreover, defect classification trained on the images augmented by the SDGAN is robust to uneven and poor lighting conditions. Note to Practitioners-This article proposes a method of defect image generation to address the lack of industrial defect images. Traditional defect recognition methods have two disadvantages: different types of defects require different algorithms and handcrafted features are deficient. Defect recognition using deep learning can solve the above problems. However, deep learning requires a plethora of images, and the number of industrial defect images cannot meet this requirement. We propose a new defect image-generation method called SDGAN to generate a defect image data set that balances diversity and authenticity. In practice, we employ a large number of defect-free images to generate a large number of defect images using our method to expand the industry defect-free image data set. Then, the augmented defect data set is used to build a deep-learning defect recognition model. Experiments show that the accuracy of defect recognition can be significantly improved by building a deep-learning defect recognition model using the augmented data set. Therefore, deep learning can achieve excellent performance in defect recognition with a limited number of defect images.

Research on small sample defect detection method based on AnoGAN and U-net

Small-scale defect detection in industrial environment based on lightweight deep learning network

Using ISU-GAN for unsupervised small sample defect detection

Dual-Path GAN: A Method for Enhancing Small-scale Defect Detection on Metal Images

Surface Defect Detection of Mobile Phone Screen Based on Generative Adversarial Networks

DefectGAN: Weakly-Supervised Defect Detection Using Generative Adversarial Network

A small sample nonstandard gear surface defect detection method

A Survey of Defect Detection Applications Based on Generative Adversarial Networks

Defect Image Sample Generation with GAN for Improving Defect Recognition

A Surface Defect Detection Method Based on Image Guided Conditional Diffusion Model

P‐4.10: Simulation Algorithm of Industrial Defects based on Generative Adversarial Network

AnomalySeg: Deep Learning-Based Fast Anomaly Segmentation Approach for Surface Defect Detection

DefectGAN: Generate Industrial Defects from Extremely Few Samples

Industrial surface defect detection by multi-scale Inpainting-GAN

Cross-Domain Defect GAN: Defect Generation for Defect Detection in Industrial Area

DG-GAN: A High Quality Defect Image Generation Method for Defect Detection

Surface defect sample generation method based on GAN

Deep-Learning-Based Small Surface Defect Detection Via an Exaggerated Local Variation-Based Generative Adversarial Network

Automated Optical Inspection Method for Light-Emitting Diode Defect Detection Using Unsupervised Generative Adversarial Neural Network

An LCD Detection Method Based on the Simultaneous Automatic Generation of Samples and Masks Using Generative Adversarial Networks

An Improved GAN Model Based on Positive Samples for LED Die Defect Detection