基于改进DCGAN轮胎缺陷图像生成方法
DOI:
CSTR:
作者:
作者单位:

1.河北科技大学信息科学与工程学院河北科技大学文法学院;2.河北科技大学信息科学与工程学院;3.贵州大学电气工程学院

作者简介:

通讯作者:

中图分类号:

基金项目:

河北省重点研发计划项目(21351801D);轨道交通关键装备智能运维平台研发(20310806D)


Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对深度卷积生成对抗网络的数据扩充方法存在生成图像质量差,模型框架不稳定,模型收敛速度慢等问题,提出一种改进DCGAN轮胎缺陷图像生成模型。将残差网络和注意力机制嵌入到DCGAN模型中,以提升模型特征的提取能力;同时摒弃DCGAN损失函数JS散度,使用带有梯度惩罚项的Wasserstein距离,以提高模型训练的稳定性。实验结果表明,使用该模型生成轮胎缺陷图像质量优于DCGAN、WGAN、CGAN与SAGAN所生成图像,其平均FID值可以达到116.28,最小FID值可以达到84.94。所提模型可以稳定地生成质量更好的轮胎缺陷图像,为轮胎缺陷样本数据集扩充提供了一种有效途径,缓解了深度学习在缺陷检测领域发展所面临的小样本问题。

    Abstract:

    参考文献
    相似文献
    引证文献
引用本文
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2023-04-28
  • 最后修改日期:2023-07-17
  • 录用日期:2023-09-18
  • 在线发布日期:
  • 出版日期:
文章二维码