wgan-gp 代码使用

在查看并对比一些代码之后，我找到了下面这个代码，其中包括了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行可以修改图片的形状，当然数据集当中的所有图片需要统一大小。