在查看并对比一些代码之后,我找到了下面这个代码,其中包括了
DCGAN,LSAGN,WGAN,以及WGAN-GP的代码
地址:
https://github.com/LynnHo/DCGAN-LSGAN-WGAN-WGAN-GP-Tensorflow
其中代码的环境要求是
但是由于我使用的python3,所以由于版本问题遇到了以下问题,
1、Argument must be a dense tensor: range(1, 4) – got shape [3], but wanted [].
原因是python2转python3后,range()返回的shape是range(0,3)而不再是list。
图片来自参考网站,
https://blog.csdn.net/ygfrancois/article/details/80688265
实际上我在使用代码 train_mnist_wgan_gp.py的时候修改的是这一行
2、数据集
readme中提到mnist的数据集在运行代码的时候会自动下载的,我在使用的时候的确自己下载了,但是不知道为什么下载的数据集没有办法正确加载,所以我使用另一个下载的数据集取代
地址:
https://github.com/bojone/gan/tree/master/MNIST_data
这个地址也是一个wgan-gp的代码,但是这个代码在使用的时候并不能自行保存模型
另外两个数据集中,卡通数据集比较小,容易下载。
我不能准确下载人脸数据集。
3、卡通数据集使用
首先,将卡通数据集在下载mnist的时候产生的data文件夹中解压,然后运行 train_cartoon_wgan_gp.py就可以了
在上面的那行代码中:
31行可以修改图片的路径以及文件夹名称,
32行可以修改图片的形状,当然数据集当中的所有图片需要统一大小。