CycleGan需要训练多少个epoch
来源:9-7 无配对图像翻译CycleGAN(2)
DATUONIAO
2020-06-13
问题描述:我尝试实战CycleGan 使用kaggle上的 horse2zebra数据集进行训练
图像尺寸:(256,256,3)
生成网络: u-net
下图是训练了99个epoch 后出现的效果
从左到右依次是:
马->马生成的斑马->马生成的斑马生成的马
斑马->斑马生成的马->斑马生成的马生成的斑马
问题:
1、出现这样的结果,是网络结果不合理,还是训练步数不够
2、一般CycleGan 网络大概需要训练多少步(大概范围)才能出结果
写回答
1回答
-
看你的训练布数是够的,一般都用不了99个epoch,二三十个epoch就可以。
你的图片大小设置的是多少?我看图变模糊了很多。一般用400x400的图像。
有没有identity loss,可以先convert回原图训练出能复原图像的GAN再去做迁移。
这块确实比较难训练,估计同学你要多费心找找其他人的经验了。
022020-06-17
相似问题