代码将上传到github上:https://github.com/Jun10ng/stackGan_with_tf1
stackGan_with_tf1
stackGan原作者的tf版(tf0.12)源码似乎有点问题,我把自己的版本贴上来(稍微修改了原作者的一些代码)。
设置路径
clone原作者的工程文件到自己电脑上,开始准备复现,代码一跑报错:
例如:
ImportError:No module named misc.datasets
在作者的README.md上有一句很容易被忽视的话
please add the project folder to PYTHONPATH
这句话的意思是,将“StackGan”的工程路径添加到~/.bashrc
export PYTHONPATH="${PYTHONPATH}:工程路径"
将“工程路径”替换成你自己的路径,然后添加到~/.bashrc中,保存。
运行 source ~/.bashrc
安装依赖和TF1.0
pip install prettytensor progressbar python-dateutil easydict pandas torchfile pyyaml scipy Pillow

本文档记录了StackGAN在TensorFlow1.0环境下的复现过程,包括设置路径、安装依赖、训练两个阶段(StageI和StageII)。作者在复现过程中发现了原版代码的问题,并提供了修正后的版本。StageI训练完成后,需要将模型文件夹重命名为StageI,以备StageII使用。若遇到CUDA和CUDNN问题,建议按照网上的教程安装CUDA 8.0和CUDNN 5.1。
最低0.47元/天 解锁文章
1万+





