Windows下使用tensorflow+StarGAN初体验

上一篇已经介绍好如何在windows环境下配置GPU版本的Tensorflow的环境，本篇来尝试在Windows环境尝试跑一跑tensorflow版本的StarGAN。

 

StarGAN的最初地址在这里，是pytorch版本的  https://github.com/yunjey/StarGAN

 

而Tensorflow的版本是国外大神对其原版进行简化的 地址在这里  https://github.com/taki0112/StarGAN-Tensorflow#men

 

 

1、下载项目，数据集

在上面给出的地址下载tensorflow版的stargan，其中项目是不含数据集的，所以需要自己单独下载。

下载的方法也有两种，这里推荐第二种。

• 命令行下载

进入对应的path后输入命令

python download.py celebA

如果报错，显示无响应则可能需要梯子或者选择手动下载。

• 手动下载

链接:  https://pan.baidu.com/s/1GBzQlRDyw3wxgxaDyMsNWA 

提取码: wtna

下载后解压到对应的项目文件中去，目录对应下图

 

 

2、尝试运行

首先在CMD中激活你自己的tensorflow环境（环境的名字自己根据本机的情况修改）

activate tensorflow_gpu

进入对应的项目path后 尝试训练

python main.py --phase train

如果失败 并提示ResourceExhaustedError 换块显存大一点的显卡吧，似乎2G显存均会报错

 

如果成功 则尝试测试训练结果 将你想要进行测试的图片放入/dataset/test文件中 结果将会在/result/中显示出来

python main.py --phase test

 

3、结果

 

这是我利用作者给出的预训练模型进行测试的结果，可以看出大体上已经有了预期的效果，但是细节上仍有偏差，有可能是本电脑的性能限制。

预训练模型地址  https://drive.google.com/open?id=1ezwtU1O_rxgNXgJaHcAynVX8KjMt0Ua-

（需要梯子）