DCGAN-Tensorflow训练自己的数据集(新手)
代码
链接:https://github.com/carpedm20/DCGAN-tensorflow
具体实践过程
1.在DCGAN-tensorflow-master中建立一个data文件夹,并在data文件夹里面建立celeA(自己的数据集名称)和mnist文件夹,其中celeA(自己的数据集名称)中存放自己的样本图片,mnist文件夹中存放四个文件(文章最后的博客链接中有一个博客有文件的下载),注意文件名称不要有个别符号的变化。四个文件主要为:
最后,文件夹顺序:
DCGAN-tensorflow-master——>data——>celebA(自己的数据集名称),mnist
celebA(自己的数据集名称)——>自己的数据集
mnist——>四个文件
2.main.py文件中的参数按照自己的需要修改,其中注意dataset名字改为自己的数据集文件夹名称,例如celeA
3.运行即可
运行时遇到的问题和解决
有一些需要安装的包自己百度进行安装即可,很多问题我后面参考的博客链接里面都有解决办法。下面我说几个我遇到的问题:
问题1: TypeError: ‘NoneType’ object is not subscriptable