wgan-gp 代码使用

在查看并对比一些代码之后,我找到了下面这个代码,其中包括了DCGAN,LSAGN,WGAN,以及WGAN-GP的代码

地址:https://github.com/LynnHo/DCGAN-LSGAN-WGAN-WGAN-GP-Tensorflow

其中代码的环境要求是

wgan-gp 代码使用_第1张图片

但是由于我使用的python3,所以由于版本问题遇到了以下问题,

1、Argument must be a dense tensor: range(1, 4) - got shape [3], but wanted [].

原因是python2转python3后,range()返回的shape是range(0,3)而不再是list。

wgan-gp 代码使用_第2张图片

图片来自参考网站,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行可以修改图片的形状,当然数据集当中的所有图片需要统一大小。

              

 

你可能感兴趣的:(wgan-gp 代码使用)