一种基于CycleGAN的接触网零部件缺陷数据生成方法及系统
摘要:
本发明公开了一种基于CycleGAN的接触网零部件缺陷数据生成方法及系统,属于接触网零部件缺陷图片处理技术领域。目前含有缺陷的接触网零部件图像非常少,远远不能满足模型训练时所需要的正负样本数量。本发明的一种基于CycleGAN的接触网零部件缺陷数据生成方法,构建CycleGAN模型,CycleGAN模型对少量的缺陷图片进行学习,获取缺陷图片的风格特征,然后将这些风格特征赋予正常图像,以形成新的缺陷图像。由于正常图片容易获得,并且数量巨大,因此利用CycleGAN模型可以轻松获得大批量的缺陷图像,方案简单,实用,切实可行;本发明能直接利用正常图片转换成缺陷图片,构思巧妙,打破现有的必须通过缺陷图像生成缺陷图像的技术偏见。
0/0