一种用于生成光伏板缺陷图像的CycleGAN模型的训练方法及装置
摘要:
本申请提供了一种用于生成光伏板缺陷图像的CycleGAN模型的训练方法及装置,该方法包括:获取训练样本,将所述训练样本输入至待训练的用于生成光伏板缺陷图像的循环生成对抗网络CycleGAN模型中,获取由所述CycleGAN模型输出的光伏板的候选图像,并根据所述训练样本和所述候选图像对所述循环生成对抗网络CycleGAN模型进行训练,以获取用于生成光伏板缺陷图像的目标CycleGAN模型。由此,本申请通过用于生成光伏板缺陷图像的目标CycleGAN模型,可以将光伏板的正常图像转换成不同类型的光伏板的缺陷图像,解决了光伏板缺陷图像较少的问题,提高了光伏板缺陷检测的准确率。
0/0