陆顺成, 姜小峰, 石奇. 基于改进DCGAN的汽车冷凝器图像生成方法[J]. 微电子学与计算机, 2022, 39(5): 71-77. DOI: 10.19304/J.ISSN1000-7180.2021.1216
引用本文: 陆顺成, 姜小峰, 石奇. 基于改进DCGAN的汽车冷凝器图像生成方法[J]. 微电子学与计算机, 2022, 39(5): 71-77. DOI: 10.19304/J.ISSN1000-7180.2021.1216
LU Shuncheng, JIANG Xiaofeng, SHI Qi. Automotive condenser image generation method based on improved DCGAN[J]. Microelectronics & Computer, 2022, 39(5): 71-77. DOI: 10.19304/J.ISSN1000-7180.2021.1216
Citation: LU Shuncheng, JIANG Xiaofeng, SHI Qi. Automotive condenser image generation method based on improved DCGAN[J]. Microelectronics & Computer, 2022, 39(5): 71-77. DOI: 10.19304/J.ISSN1000-7180.2021.1216

基于改进DCGAN的汽车冷凝器图像生成方法

Automotive condenser image generation method based on improved DCGAN

  • 摘要: 深度学习方法在工业产品图像缺陷检测领域取得进展,但是大量的缺陷数据难以收集.针对在生成汽车冷凝器缺陷图像过程中存在生成质量低、无法按缺陷类别生成,模型收敛缓慢等问题,将生成对抗网络应用于缺陷图像的生成,提出了一种基于半监督和自注意力机制的深度卷积生成对抗网络(DCGAN)模型用于生成汽车冷凝器外观缺陷图像.在DCGAN中引入自注意力机制,克服卷积网络长距离特征提取的问题,提高了生成样本的质量;通过半监督学习,在无监督判别器中加入监督辅助分类器,并将分类器的交叉熵损失和梯度惩罚加入到判别器的损失函数中,提高了模型的收敛速度和稳定性;使用条件归一化调整卷积层参数,并将图像的缺陷类别信息嵌入到判别器中,提高了生成样本的多样性,使得模型能够生成特定缺陷的冷凝器图像.实验结果表明,所提出的模型能够生成高质量的汽车冷凝器缺陷图像,fid值达到了43.7,优于现有的DCGAN和SAGAN。与ACGAN相比,生成图像的多样性相比也有明显提高.

     

    Abstract: Deep learning methods have made progress in the field of industrial product image defect detection, but it is difficult to collect a large number of defect data. The existing methods have problems on generating automotive condenser defect images, such as low image generation quality, inability to generate images according to defect categories, and slow model convergence speed. Applying generative adversarial network to defect image generation, adeep convolution generative adversarial network (DCGAN) model based on semi-supervised and self-attention mechanism is proposed togenerateautomotivecondenserdefectimages. Self-Attention mechanism is introduced in DCGAN to overcome the problem of long-range feature extraction in convolutional network and improve the quality of generated samples. By semi-supervised learning, a supervised classifier is added into the unsupervised discriminator, the cross entropy loss of the classifier and gradient penalty are added into the loss function of the discriminator, which improves the convergence speed and stability of the model. Condition normalization is used to adjust convolution layer parameters, and the defect category information of images is embedded into the discriminator, which improves the diversity of generated samples and enable the model to generate condenser images with specific defects. Experimental results show that the proposed method can generate high-quality automotive condenser defect images, the FID value reaches 43.7, which is better than the existing DCGAN and SAGAN. Compared with ACGAN, the diversity of generated images is also significantly improved.

     

/

返回文章
返回