Abstract:Automated surface inspection (ASI) is an important research content in computer vision. In recent years, with the application of deep learning models represented by convolutional neural networks (CNN) in computer vision, surface defect detection based on computer vision has made impressive progress. However, compared with a few or dozens of pictures in real industrial scenes, traditional deep learning methods require a large amount of annotation data for training, so it is quite difficult to adapt to the complex industrial scenarios with varying surface profiles, lighting conditions, imaging angles and environments. Despite these difficulties, there is still a wide gap in the performance between the detection of small-scale and large-scale objects. Traditional Generative Adversarial Nets (GANs) based augmentation method can only be used for classification networks or unsupervised learning, and if applied to detection networks, they need to be labeled, which requires much labor and time. In order to solve these problems, a metal surface small-scale defect images augmentation method is proposed. Our method contains two parts: generation part and augmentation part. In the generation part, two pairs of generators and discriminators are used to generate the defect areas and the background areas of the image, which can not only generate more realistic defect images than others but also save the training process for surface images. In the augmentation part, the real defect images and the generated defect images are concatenated before copy-pasting them on background images to generate annotated dataset for training. We conduct experiments on our metal surface dataset. The experimental results show that our method can generate high-quality defect samples and background samples, which greatly enriches the original dataset. We evaluate different augmentation strategies, and ultimately, we achieve 6.2% improvement on baseline and 4.4% on copy-pasting method.

Synthetic Image Augmentation for Improved Classification using Generative Adversarial Networks

Improving the Performance of Fine-Grain Image Classifiers via Generative Data Augmentation

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

Data Augmentation for Image Classification using Generative AI

The Imaginative Generative Adversarial Network: Automatic Data Augmentation for Dynamic Skeleton-Based Hand Gesture and Human Action Recognition

Detection of AI-Generated Synthetic Images with a Lightweight CNN

Enhancement of Image Classification Using Transfer Learning and GAN-Based Synthetic Data Augmentation

Performance Study of Image Data Augmentation by Generative Adversarial Networks

Synthetic Data for Object Classification in Industrial Applications

Learning to Generate Synthetic Data via Compositing

Improving synthetic media generation and detection using generative adversarial networks

The Synthetic Aperture Radar-based Image Augmentation Using Generative Adversarial Network

Data Augmentation via Mixed Class Interpolation using Cycle-Consistent Generative Adversarial Networks Applied to Cross-Domain Imagery

Defense-GAN: Protecting Classifiers Against Adversarial Attacks Using Generative Models

Semantically Controllable Augmentations for Generalizable Robot Learning

Deep Image Fingerprint: Towards Low Budget Synthetic Image Detection and Model Lineage Analysis

Improving Android Malware Detection Through Data Augmentation Using Wasserstein Generative Adversarial Networks

Analyzing Effects of Fake Training Data on the Performance of Deep Learning Systems

Anomaly Detection in Additive Manufacturing Processes using Supervised Classification with Imbalanced Sensor Data based on Generative Adversarial Network

Diffusion-based Data Augmentation for Skin Disease Classification: Impact Across Original Medical Datasets to Fully Synthetic Images

GAN Augmentation: Augmenting Training Data using Generative Adversarial Networks