animegan2-pytorch/README.md

24 lines
882 B
Markdown
Raw Normal View History

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"> &nbsp;
<img src="./samples/compare/2.jpg" width="960"> &nbsp;
<img src="./samples/compare/3.jpg" width="960"> &nbsp;
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)