返回
Diversifying Tire-Defect Image Generation Based on Generative Adversarial Network
DOI:10.1109/TIM.2022.3160542.png)
摘要
En 中文
With the development of data-driven models, deep learning has been increasingly applied in the field of defect detection. However, the performance of deep learning models is greatly restricted by costly labeling and sample scarcity. One of the best approaches to solve the data imbalance problem is increasing the quantity and diversity of defect samples. Meanwhile, the current models based on generative adversarial network (GAN) cannot readily control the category and shape of generated defect samples, which results in inefficient data augmentation. Thus, to simultaneously achieve the category and shape adjustability of defect samples, a novel model named diversifying tire-defect GAN (DTD-GAN) is proposed in this article by integrating the advantages of latent space decomposition and feature decoupling algorithm. In this model, an auxiliary classifier is designed to control the defect category by discrete latent codes, and the mutual information maximization approach is adopted to diversify the tire-defect shape via continuous latent codes that greatly improve the diversity of the generated tire-defect samples. To illustrate the effectiveness of the proposed DTD-GAN model, extensive experiments on image quality and diversity assessment are carried out on the images generated by this model and other state-of-the-art approaches, and the achieved improvement in deep learning model performance in a tire-defect detection task is also discussed.
Keyword:
Tires
Generative adversarial networks
Feature extraction
Deep learning
Shape
Data models
Training
Data augmentation
feature decoupling
generative adversarial network (GAN)
generative control
tire-defect detection
期刊
IF:
5.9
论文数:
2.0W
被引数:
5.8W
机构
引用论文
Using Deep Learning to Detect Defects in Manufacturing: A Comprehensive Survey and Current Challenges使用深度学习检测制造中的缺陷: 综合调查和当前挑战
MATERIALS
IF3.2
Mechanisms and Dynamics of Layered Structure Formation During Co-Deposition of Binary Compound Thin Films
Coatings
IF0
An End-to-End Steel Surface Defect Detection Approach via Fusing Multiple Hierarchical Features融合多个层次特征的端到端钢表面缺陷检测方法

