StackGan之TensorFlow1.0复现

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

代码将上传到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

 
  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值