AnimeGANv2照片动漫化转换工具

AnimeGANv2 是 AnimeGAN 的改进版本，是一个可以讲风景照片/视频转换为动漫图片的工具。

特性：

AnimeGANv2 的改进方向主要包括以下 4 点：

• 解决生成图像中高频伪影的问题。
• 易于训练，直接达到论文中的效果。
• 进一步减少生成器网络的参数数量。
• 使用新的高质量风格数据，尽可能来自 BD 电影。

要求

• Python 3.6
• tensorflow-gpu 1.15.0（GPU 2080Ti，cuda 10.0.130，cudnn 7.6.0）
• opencv
• tqdm
• numpy
• glob
• argparse
• onnxruntime

用法

1. 推理

python test.py --checkpoint_dir checkpoint/generator_Hayao_weight --test_dir dataset/test/HR_photo --save_dir Hayao/HR_photo

2. 将视频转换为动画

python video2anime.py --video video/input/お花見.mp4 --checkpoint_dir checkpoint/generator_Hayao_weight --output video/output

3. 训练

1. 下载 vgg19

vgg19.npy

2. 下载 Train/Val Photo 数据集

Link

 

3. edge_smooth

python edge_smooth.py --dataset Hayao --img_size 256

4. 训练

python train.py --dataset Hayao --epoch 101 --init_epoch 10

5. 提取生成器的权重

python get_generator_ckpt.py --checkpoint_dir ../checkpoint/AnimeGANv2_Shinkai_lsgan_300_300_1_2_10_1 --style_name Shinkai