2021-02-16 13:00:00 +01:00
## PyTorch Implementation of [AnimeGANv2](https://github.com/TachibanaYoshino/AnimeGANv2)
2021-02-16 12:59:02 +01:00
2021-02-17 00:32:21 +01:00
**Weight Conversion (Optional, requires TensorFlow 1.x)**
2021-02-16 12:59:02 +01:00
```
git clone https://github.com/TachibanaYoshino/AnimeGANv2
python convert_weights.py
```
**Inference**
```
python test.py --input_dir [image_folder_path]
```
**Results from converted [[Paprika ](https://drive.google.com/file/d/1K_xN32uoQKI8XmNYNLTX5gDn1UnQVe5I/view?usp=sharing )] style model**
(input image, original tensorflow result, pytorch result from left to right)
2021-02-16 13:00:24 +01:00
< img src = "./samples/compare/1.jpg" width = "960" >
< img src = "./samples/compare/2.jpg" width = "960" >
< img src = "./samples/compare/3.jpg" width = "960" >
2021-02-16 12:59:02 +01:00
2021-02-18 05:38:43 +01:00
**Note:** Training code not included / Tested on RTX3090 + PyTorch1.7.1 / Results slightly different due to the [bilinear upsample issue ](https://github.com/pytorch/pytorch/issues/10604 )