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
|
|
|
|
|
|
|
|
|
|
|
**Weight Conversion (Optional)**
|
|
|
|
```
|
|
|
|
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)
|
|
|
|
|
|
|
|
<img src="./samples/compare/1.jpg" width="650">
|
|
|
|
<img src="./samples/compare/2.jpg" width="650">
|
|
|
|
<img src="./samples/compare/3.jpg" width="650">
|
|
|
|
|
2021-02-16 12:59:47 +01:00
|
|
|
**Note:** Training code not included / Results looks slightly blurrier than the original ones.
|