项目本人: https://github.com/carpedm20/DCGAN-tensorflow
大佬的个人博客: https://carpedm20.github.io/
讲真,不管是啥gan,最适合运行这些项目的都是Linux,更推荐Linux。
一.配置虚拟环境
在annaconda中新建一个虚拟环境,然后加载环境包:
pip install Tensorflow==1.9.0
pip install SciPy==1.0.0
pip install pillow
pip install tqdm
pip install opencv-python
与作者给出的版本有所不同,tensorflow我也不知道哪里来的0.12.1,无中生有为难人嘛。scipy,因为版本更新,有很多函数已经被弃用,如果用新版本会报下面的错,所以要使用旧版本
二.准备数据集
官方readMe提供下载mnist celebA俩数据集的方法,不管我是不是科学上网都没下载下来,总是报错
所以我,直接百度mnist数据集,找了一个百度云就直接用了。but,如果直接下载mnist,一般是压缩包,把它们解压后使用,会发现会报错
当我去搜了他人博客发现是因为windows解压会造成一些问题,然后去下载了这个博主提供的在Linux下解压的mnist数据集。
链接:提取码:1024
将数据集放入data文件夹:
三.开始训练
python main.py --dataset mnist --input_height=28 --output_height=28 --train
结果在out文件夹:
模型:
训练过程图片:
四.推理测试
python main.py --dataset mnist --input_height=28 --output_height=28 --visualize=True --checkpoint_dir="D:\workspaceForMine\ForGan\ForDCGan\DCGAN-tensorflow-master\out\20220205.140627 - data - mnist - x28.z100.uniform_signed.y28.b64\checkpoint"
结果也在out文件夹里:
五.结语
讲真这个小项目它老人家不难的,但是里面老多坑了。本来我感叹自己菜,后来一看作者好像是韩国人,emmmmmm,啊,那到也不只是我菜蛤
the end