Colab试跑AnimeGAN

今天看到机器之心上的文章 AnimeGAN，感觉效果还不错，自己试一下

我用的是Google Colab，免费的，只能用一个GPU。。。【穷】

代码 

Git Clone

!git clone https://github.com/TachibanaYoshino/AnimeGAN

成功： 

 看下所在目录

import os
os.chdir('AnimeGAN')
print(os.getcwd())

 下面一步有点复杂

首先，我们要创建一个文件

然后把一下内容复制进去，保存

URL=https://github.com/TachibanaYoshino/AnimeGAN/releases/download/Haoyao-style_V1.0/Haoyao-style.zip
ZIP_FILE=./checkpoint/Haoyao-style.zip
TARGET_DIR=./checkpoint/saved_model

mkdir -p ./checkpoint
wget -N $URL -O $ZIP_FILE
mkdir -p $TARGET_DIR
unzip $ZIP_FILE -d $TARGET_DIR
rm $ZIP_FILE

DatesetURL=https://github.com/TachibanaYoshino/AnimeGAN/releases/download/dataset-1/dataset.zip
ZIP_FILE=./dataset.zip
TARGET_DIR=./dataset

rm -rf dataset
wget -N $DatesetURL -O $ZIP_FILE
unzip $ZIP_FILE -d $TARGET_DIR
rm $ZIP_FILE

VGG_FILE=./vgg19_weight/vgg19.npy
wget --load-cookies /tmp/cookies.txt "https://docs.google.com/uc?export=download&confirm=$(wget --quiet --save-cookies /tmp/cookies.txt --keep-session-cookies --no-check-certificate 'https://docs.google.com/uc?export=download&id=1U5HCRpZWAbDVLipNoF8t0ZHpwCRX7kdF' -O- | sed -rn 's/.*confirm=([0-9A-Za-z_]+).*/\1\n/p')&id=1U5HCRpZWAbDVLipNoF8t0ZHpwCRX7kdF" -O $VGG_FILE && rm -rf /tmp/cookies.txt

 然后运行命令

!bash download_staffs.sh

出现如下内容说明成功：

 下面开始训练

我们要利用Colab GPU先做如下修改：

然后，选择GPU：

成功分配资源如图所示：

然后，我们需要看一下GPU编号：

用如下命令：

！nvidia-smi

可以看到，编号为0 

然后我们需要修改文件：

修改如下：

运行命令：

!python main.py --phase train --dataset Hayao --epoch 1 --init_epoch 1

我把epoch调成了1，时间仓促，想要效果好的多弄几个epoch。

训练成功如图所示：

 Finish！开始测试！

!python test.py --checkpoint_dir checkpoint/saved_model --test_dir dataset/test/real --style_name H

成功后显示如下：

测试图像在路径：

输出图像在路径：

 大功告成！

 这里补充说明一点：

1. 如果想保存我们的文件夹，需要将其放到Drive->My Drive目录下。这样下次打开的时候就可以看到了。

2. checkpoint/AnimeGAN_Hayao_lsgan_300_300_1_3_10文件夹下是自己训练的模型。

3.附上最新的测试效果，发现风景图片的测试效果还是不错的！