Defective samples simulation through Neural Style Transfer for automatic surface defect segment

10/08/2019
by   Taoran Wei, et al.
9

Owing to the lack of defect samples in industrial product quality inspection, trained segmentation model tends to overfit when applied online. To address this problem, we propose a defect sample simulation algorithm based on neural style transfer. The simulation algorithm requires only a small number of defect samples for training, and can efficiently generate simulation samples for next-step segmentation task. In our work, we introduce a masked histogram matching module to maintain color consistency of the generated area and the true defect. To preserve the texture consistency with the surrounding pixels, we take the fast style transfer algorithm to blend the generated area into the background. At the same time, we also use the histogram loss to further improve the quality of the generated image. Besides, we propose a novel structure of segment net to make it more suitable for defect segmentation task. We train the segment net with the real defect samples and the generated simulation samples separately on the button datasets. The results show that the F1 score of the model trained with only the generated simulation samples reaches 0.80, which is better than the real sample result.

READ FULL TEXT

page 2

page 6

page 10

research
04/30/2020

Contextual Text Style Transfer

We introduce a new task, Contextual Text Style Transfer - translating a ...
research
02/08/2023

Neural Artistic Style Transfer with Conditional Adversaria

A neural artistic style transformation (NST) model can modify the appear...
research
08/03/2022

Style Transfer of Black and White Silhouette Images using CycleGAN and a Randomly Generated Dataset

CycleGAN can be used to transfer an artistic style to an image. It does ...
research
01/06/2023

Tackling Data Bias in Painting Classification with Style Transfer

It is difficult to train classifiers on paintings collections due to mod...
research
02/17/2020

Neural arbitrary style transfer for portrait images using the attention mechanism

Arbitrary style transfer is the task of synthesis of an image that has n...
research
03/24/2022

Industrial Style Transfer with Large-scale Geometric Warping and Content Preservation

We propose a novel style transfer method to quickly create a new visual ...
research
10/29/2021

Learning Co-segmentation by Segment Swapping for Retrieval and Discovery

The goal of this work is to efficiently identify visually similar patter...

Please sign up or login with your details

Forgot password? Click here to reset