MINZU2018 / stylegan_reimplementation

Reimplementation of https://arxiv.org/abs/1812.04948

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

This is a reimplementation of NVidia's stylegan https://github.com/NVlabs/stylegan I did for learning purposes. My priority was testing out changes such as non-square images and conditioning on labels (using acgan and projection discriminator) rather than keeping the code clean so right now it is a bit messy.

This has been tested, and supports features like rectangular images, but I am currently waiting on the most recent training session to finish before uploading results. Training and sample generation are both performed using 'run.py'.

It is mostly original, but includes a couple functions from the official implementation for comparison testing.

To run, see the comments in start_training.sh (and then run start_training.sh). This has only been tested in one environment, so feel free to create a github issue if you encounter a problem. This model can handle non-square images and I plan on organizing and including the tools I've made/modified to generate datasets like that soonish.

Blogs related to a tool I made to interact with StyleGAN models:

https://towardsdatascience.com/animating-ganime-with-stylegan-part-1-4cf764578e https://towardsdatascience.com/animating-ganime-with-stylegan-the-tool-c5a2c31379d

Original Paper: Karras, T., Laine, S., and Aila, T. A style-based generator architecture for generative adversarial networks

About

Reimplementation of https://arxiv.org/abs/1812.04948

License:Apache License 2.0


Languages

Language:Python 98.7%Language:Shell 1.3%