us

unet-stylegan2

A Pytorch implementation of Stylegan2 with UNet Discriminator

Showing:

Popularity

Downloads/wk

0

GitHub Stars

149

Maintenance

Last Commit

10mos ago

Contributors

2

Package

Dependencies

8

License

GPLv3+

Categories

Readme

UNet Stylegan2

An implementation of Stylegan2 with UNet Discriminator. This repository works largely the same way as Stylegan2 Pytorch. Simply replace all the stylegan2_pytorch command with unet_stylegan2 instead.

Update: Results have been very good. Will need to investigate combining this with a few other techniques, and then I will write up full instructions for use.

Install

$ pip install unet-stylegan2

Usage

$ unet_stylegan2 --data ./path/to/data

Citations

@misc{karras2019analyzing,
    title={Analyzing and Improving the Image Quality of StyleGAN},
    author={Tero Karras and Samuli Laine and Miika Aittala and Janne Hellsten and Jaakko Lehtinen and Timo Aila},
    year={2019},
    eprint={1912.04958},
    archivePrefix={arXiv},
    primaryClass={cs.CV}
}
@misc{schnfeld2020unet,
    title={A U-Net Based Discriminator for Generative Adversarial Networks},
    author={Edgar Schönfeld and Bernt Schiele and Anna Khoreva},
    year={2020},
    eprint={2002.12655},
    archivePrefix={arXiv},
    primaryClass={cs.CV}
}

Rate & Review

Great Documentation0
Easy to Use0
Performant0
Highly Customizable0
Bleeding Edge0
Responsive Maintainers0
Poor Documentation0
Hard to Use0
Slow0
Buggy0
Abandoned0
Unwelcoming Community0
100