3.3 KiB
3.3 KiB
PyTorch Implementation of AnimeGANv2
Updates
2020-10-17Add weights for face portrait v22020-11-07Thanks to ak92501, a web demo is integrated to Huggingface Spaces with Gradio. See demo:
Weight Conversion from the Original Repo (Requires TensorFlow 1.x)
git clone https://github.com/TachibanaYoshino/AnimeGANv2
python convert_weights.py
Inference
python test.py --input_dir [image_folder_path] --device [cpu/cuda]
Results from converted [Paprika] style model
(input image, original tensorflow result, pytorch result from left to right)
Note: Training code not included / Tested on RTX3090 + PyTorch1.7.1 / Results from converted weights slightly different due to the bilinear upsample issue
Additional Models
Webtoon Face [ckpt]
samples
Trained on 256x256 face images. Distilled from webtoon face model with L2 + VGG + GAN Loss and CelebA-HQ images. See test_faces.ipynb for details.
Face Portrait v1 [ckpt]
Face Portrait v2 [ckpt]
samples
Trained on 512x512 face images. Compared to v1, 🔻beautify 🔺robustness
🦑 🎮 🔥
