使用keras训练cyclegan

cyclegan虽然已经有一定的时间了,不算比较新的对抗生成网络,但是还是拿来玩玩还是可以的,这里我是直接用的人家写好的代码,拿来训练的,代码在这里

值得注意的是,这里进行训练的时候需要手动修改data_loader.py里面的path_A和path_B为绝对路径好一些,因为相对路径有可能会导致加载不到数据,我反正是遇到了这个坑。

具体需要用到的数据集在这里,我已经下载好了

1)monet2photo(5vf0 )
复制这段内容后打开百度网盘手机App,操作更方便哦

2)apple2orange(或者这里4h3v)

3)ma2banma(或者这里49wd)

我自己训练好了200代,但是自己只保存到了150代的权重,这里是保存的权重值(有一个是20,有一个是10, 有一个是150)

1)monet2photo(t9fj) 

2)apple2orange(ao07)

3)ma2banma (gecz)

看一下效果:

使用keras训练cyclegan