一种基于循环生成对抗网络的弧面缺陷图像生成方法

作者：黄茜; 朱轲信; 胡志辉; 师聪颖

来源：2021-03-09, 中国, CN202110253845.1.

本发明属于缺陷图像生成技术领域,涉及一种基于循环生成对抗网络的弧面缺陷图像生成方法,包括：步骤1、构建缺陷集和无缺陷集；步骤2、初始化循环生成对抗网络；循环生成对抗网络包括用于从无缺陷的弧面图像转换为含缺陷的弧面图像的GAN-(B→A)和用于从含缺陷的弧面图像转换为无缺陷的弧面图像的GAN-(A→B),GAN-(B→A)和GAN-(A→B)网络结构完全一致；步骤3、训练循环生成对抗网络；步骤4、基于训练好的循环生成对抗网络将无缺陷的弧面图像生成含缺陷的弧面图像。本发明实现了弧面缺陷图像的快速生成,生成缺陷的形状以及面积具有随机性,仅需前期少量的标注工作就可以实现大量缺陷样本的自动化生成。

• 单位
  华南理工大学