StackGan之TensorFlow1.0复现
发布于2019-08-07 11:57 阅读(1024) 评论(0) 点赞(2) 收藏(1)
代码将上传到github上：https://github.com/Jun10ng/stackGan_with_tf1
stackGan_with_tf1
stackGan原作者的tf版（tf0.12）源码似乎有点问题，我把自己的版本贴上来（稍微修改了原作者的一些代码）。
设置路径
clone原作者的工程文件到自己电脑上，开始准备复现，代码一跑报错：
例如：
ImportError：No module named misc.datasets
- 1
在作者的README.md上有一句很容易被忽视的话
please add the project folder to PYTHONPATH
- 1
这句话的意思是，将“StackGan”的工程路径添加到~/.bashrc
export PYTHONPATH="${PYTHONPATH}:工程路径"
- 1
将“工程路径”替换成你自己的路径，然后添加到~/.bashrc中，保存。
运行 source ~/.bashrc
安装依赖和TF1.0
pip install prettytensor progressbar python-dateutil easydict pandas torchfile pyyaml scipy Pillow
- 1
安装tf1.0
pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-0.12.0rc1-cp27-none-linux_x86_64.whl
- 1
训练
这部分和原作者的就一模一样了
StageI
python stageI/run_exp.py --cfg stageI/cfg/birds.yml --gpu 0
- 1
训练完StageI后，将得到的模型的父文件夹改名为StageI(原本的名字是系统按照时间自动命名的，如StageI_2019_0625…)
如果有报错关于CUDA和CUDNN的，请按网上教程安装CUDA（8.0）和CUDNN（5.1）
StageII
python stageII/run_exp.py --cfg stageII/cfg/birds.yml --gpu 0
- 1
所属网站分类: 技术文章 > 博客
作者：python好学吗
链接：https://www.pythonheidong.com/blog/article/10536/6be11752d33375993f92/
来源：python黑洞网
任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任
昵称:
评论内容：(最多支持255个字符)
---无人问津也好，技不如人也罢，你都要试着安静下来，去做自己该做的事，而不是让内心的烦躁、焦虑，坏掉你本来就不多的热情和定力