wgan-gp 代码使用

  • Post author:
  • Post category:其他


在查看并对比一些代码之后,我找到了下面这个代码,其中包括了

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



版权声明:本文为Diana_Z原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。