AnimeGAN照片轉(zhuǎn)化為動漫圖

AnimeGAN 使用 GAN 框架將現(xiàn)實(shí)世界的照片轉(zhuǎn)換為動漫圖像。

以下是需要注意的事項(xiàng)：

1. 由于訓(xùn)練集中的真實(shí)照片都是風(fēng)景照片，如果你想對以人物為主體的照片進(jìn)行風(fēng)格化，最好在訓(xùn)練集中添加至少 3000 張人物照片并重新訓(xùn)練，以獲得新的模型。
2. 為了獲得更好的人臉動畫效果，在使用 2 張圖片作為數(shù)據(jù)對進(jìn)行訓(xùn)練時，建議照片中的人臉和動漫風(fēng)格數(shù)據(jù)中的人臉在性別上盡可能保持一致。
3. 生成的風(fēng)格化圖片會受到風(fēng)格數(shù)據(jù)整體亮度和色調(diào)的影響，所以盡量不要選擇夜晚的動漫圖片作為風(fēng)格數(shù)據(jù)，需要對整體風(fēng)格數(shù)據(jù)進(jìn)行曝光補(bǔ)償，以提升整個風(fēng)格數(shù)據(jù)的明暗一致性。

AnimeGANv2，AnimeGAN 的改進(jìn)版本已經(jīng)發(fā)布。

依賴項(xiàng)

• python 3.7
• tensorflow-gpu 1.15.0 (ubuntu, GPU 2080Ti, cuda 10.0.130, cudnn 7.6.0)
• opencv
• tqdm
• numpy
• glob
• argparse

用法

推理

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

將視頻轉(zhuǎn)換為動畫

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

訓(xùn)練

1. 下載 vgg19 或 Pretrained 模型

• vgg19.npy
• 預(yù)訓(xùn)練模型

2. 下載數(shù)據(jù)集

• 關(guān)聯(lián)

3. edge_smooth

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

4.訓(xùn)練

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

5.提取生成器的權(quán)重

python get_generator_ckpt.py --checkpoint_dir ../checkpoint/AnimeGAN_Hayao_lsgan_300_300_1_1_10 --style_name Hayao