Paper
12 October 2022 Surface defect sample generation method based on GAN
Fangyi Ni, Xiaojun Wu, Jinghui Zhou, Zhichang Liu
Author Affiliations +
Proceedings Volume 12342, Fourteenth International Conference on Digital Image Processing (ICDIP 2022); 123421N (2022) https://doi.org/10.1117/12.2644555
Event: Fourteenth International Conference on Digital Image Processing (ICDIP 2022), 2022, Wuhan, China
Abstract
In order to solve the insufficiency of training data when deep learning technology is applied to surface defect detection task, a surface defect generation algorithm based on generative adversarial network (GAN) was proposed to enhance training sample data. First, a U-shaped convolutional network was designed, and a spatial adaptive normalized structure was introduced to control the mask image to generate the defect shape, and the network from defect-free image to defect image was completed. Second, a multi-layer convolutional discriminant network is designed to extract adversarial feature of the real samples and generated samples. Finally, the adversarial training loss was designed and the generative network adversarial training was completed. Through quantitative contrast experiment, it is proved that the segmentation network has better segmentation results than without data augmentation after using the surface defect generation algorithm to generate data for data augmentation.
© (2022) COPYRIGHT Society of Photo-Optical Instrumentation Engineers (SPIE). Downloading of the abstract is permitted for personal use only.
Fangyi Ni, Xiaojun Wu, Jinghui Zhou, and Zhichang Liu "Surface defect sample generation method based on GAN", Proc. SPIE 12342, Fourteenth International Conference on Digital Image Processing (ICDIP 2022), 123421N (12 October 2022); https://doi.org/10.1117/12.2644555
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Gallium nitride

LCDs

Convolution

Defect detection

Image segmentation

Network architectures

Image quality

RELATED CONTENT


Back to Top