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