maxmustermann/animegan2-pytorch
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity
16 commits 1 branch 0 tags 38 MiB
Jupyter Notebook 99.1%
Python 0.9%
Go to file
sr9 09cc72dad7 additional cli
2021-03-03 19:44:57 +09:00
samples upsampling method changed + face model added 2021-02-21 15:04:52 +09:00
.gitignore Initial commit 2021-02-16 20:34:22 +09:00
convert_weights.py typo fixed 2021-02-17 14:03:56 +09:00
model.py additional cli 2021-03-03 19:44:57 +09:00
README.md Update README.md 2021-02-23 07:47:03 +09:00
test.py additional cli 2021-03-03 19:44:57 +09:00
test_faces.ipynb upsampling method changed + face model added 2021-02-21 15:04:52 +09:00

README.md

PyTorch Implementation of AnimeGANv2

Face Model: Distilled from this model with L2 + VGG + GAN Loss and CelebA-HQ images. See test_faces.ipynb for the inference. Model file can be downloaded from here (8MB). Enjoy!

 

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