Generative adversarial networks to improve the robustness of visual defect segmentation by semantic networks in manufacturing components

FA Saiz, G Alfaro, I Barandiaran, M Graña - Applied Sciences, 2021 - mdpi.com
Applied Sciences, 2021mdpi.com
This paper describes the application of Semantic Networks for the detection of defects in
images of metallic manufactured components in a situation where the number of available
samples of defects is small, which is rather common in real practical environments. In order
to overcome this shortage of data, the common approach is to use conventional data
augmentation techniques. We resort to Generative Adversarial Networks (GANs) that have
shown the capability to generate highly convincing samples of a specific class as a result of …
This paper describes the application of Semantic Networks for the detection of defects in images of metallic manufactured components in a situation where the number of available samples of defects is small, which is rather common in real practical environments. In order to overcome this shortage of data, the common approach is to use conventional data augmentation techniques. We resort to Generative Adversarial Networks (GANs) that have shown the capability to generate highly convincing samples of a specific class as a result of a game between a discriminator and a generator module. Here, we apply the GANs to generate samples of images of metallic manufactured components with specific defects, in order to improve training of Semantic Networks (specifically DeepLabV3+ and Pyramid Attention Network (PAN) networks) carrying out the defect detection and segmentation. Our process carries out the generation of defect images using the StyleGAN2 with the DiffAugment method, followed by a conventional data augmentation over the entire enriched dataset, achieving a large balanced dataset that allows robust training of the Semantic Network. We demonstrate the approach on a private dataset generated for an industrial client, where images are captured by an ad-hoc photometric-stereo image acquisition system, and a public dataset, the Northeastern University surface defect database (NEU). The proposed approach achieves an improvement of 7% and 6% in an intersection over union (IoU) measure of detection performance on each dataset over the conventional data augmentation.
MDPI
以上显示的是最相近的搜索结果。 查看全部搜索结果