最近需要跑DCGAN的python项目,代码在这里:
carpedm20/DCGAN-tensorflow 的github地址
在自己电脑上跑这个程序时遇到了几个问题:
1、我电脑上装了pycharm+python3.6,一般的python程序直接在pycharm里面运行很方便,但这个项目并不能直接在pycharm里面用“Run”来运行,因为download.py和main.py在执行的时候都需要输入参数。所以只能通过python命令来执行。
2、python命令的执行方式就有多种了:
- 直接通过python命令窗口进行运行;
- 如果Python环境配好了可以通过cmd命令窗口来执行。要注意先进入项目文件的目录,再执行python命令;
- 如果装有git,也可以打开项目目录,然后在空白处右键,选择“Git Bash Here”,打开git窗口,直接输入python执行语句。
3、mnist和celebA数据集下载不下来。原项目里面的通过执行download.py文件进行下载,但可能由于外网限制或是网速不好,使用命令下载不好使。这时候就考虑自己手动下载了。
- mnist数据集可以在 http://yann.lecun.com/exdb/mnist/ 下载,四个.gz文件都下载。之后再原项目的data文件夹里面新建一个mnist文件夹,把下载好的这四个文件都放进去。但是我发现其实这样还是无法运行,提示说是里面的文件识别不了:
有一篇博客说是这个文件需要在Linux环境下解压才可以,我没有linux环境所以这个办法还没有尝试。 - celebA数据集同样是用命令下载不下来。在网上找到了资源,是一个里面包含.jpg文件的超大压缩包,,文件真的挺大的。
有需要celebA文件链接的可以看这里:
链接:https://pan.baidu.com/s/1FaFlezxF-Q6Cf9KX65Jxqw 密码:bu5y
注意:下载之后记得解压,记得把名字改成celebA,记得把它放在data文件夹下。
- 4、然后就是用celebA数据集进行测试了。执行命令:
python main.py --dataset celebA --input_height=108 --train --crop,然后就是漫长的等待过程。运行的截图是这样子的:
另一个博主用 自己从网上爬取的动漫人物图像,并且做了处理,数据集相比celeBA更小一些,过程也介绍的非常详细,地址是:
http://geek.csdn.net/news/detail/134553
只是有一点,他给的运行命令我运行有错,稍微改一下就能跑了。
作者原命令是: python main.py --image_size 96 --output_size 48 --dataset anime --is_crop True --is_train True --epoch 300
我改动的是: $ python main.py --image_size 96 --output_size 48 --dataset anime --crop --train --epoch 300
其实就是把crop和train稍微稍微改了一下,因为我看源码感觉因为用我改的这种写法。
两个数据集我都跑了,都可以跑起来,只是,,嗯,电脑太太太太太慢了,我没有等到最后就放弃了。
mnist用了linux进行了解压好像还是有点问题,已经放弃这个数据集了。