TachibanaYoshino / AnimeGAN

A Tensorflow implementation of AnimeGAN for fast photo animation ! This is the Open source of the paper 「AnimeGAN: a novel lightweight GAN for photo animation」, which uses the GAN framwork to transform real-world photos into anime images.
4.46k stars 660 forks source link

Pytorch implemenation #51

Open ptran1203 opened 3 years ago

ptran1203 commented 3 years ago

Hi @TachibanaYoshino,

Great work! Your project is really interesting, so I decided to implement the algorithm in Pytorch based on your Tensorflow code.

I tried to make the architecture of G and D to be the same as yours, and train on the same config (loss weights) but the result is a little bit different. So I tuned the loss weights to get the best result.

Currently, my results look like anime, but some generation images still have some artifacts. Maybe I will need to train longer.

Could you add this implementation into your README.md, I think It helps other people when looking for a Pytorch version :smile:

thank you very much.

my repo: https://github.com/ptran1203/pytorch-animeGAN

Some of my results:

1 (46)_anime 1 (36)_anime

TachibanaYoshino commented 3 years ago

Ok, I'll add it to the readme later.

ptran1203 commented 3 years ago

Thank you!