项目及数据地址
CycleGAN项目的github地址为:GitHub - junyanz/CycleGAN: Software that can generate photos from paintings, turn horses into zebras, perform style transfer, and more.其中horse2zebra数据集下载链接:链接:https://pan.baidu.com/s/1FXJ6AhD_GmHBsNAYTXReTA
提取码:wfka
horse2zebra数据集
下载好horse2zebra数据集后里面有4个文件夹,如下图所示。
打开trainA文件夹后的数据样式为注意,在跑给出的demo时,要提前安装好可视化的第三方库visdom(pip install visdom),在跑程序前记得在cmd中使用命令python -m visdom.server启动visdom,否则会报如下错误:
运行1个epoch后结果
运行1个epoch后的可视化结果如下,看右边最上面那4张图,第一幅是真实输入的图片(real_A);第二幅由生成器(A->B)生成的假的图片(fake_B);第三幅图(rec_A)是由第二幅图(fake_B)还原后的图,要求与real_A尽可能一致;第四幅图(idt_B)表示由real_A生成的A风格的图。第二排图的解释同上。