/unet-stylegan2

A Pytorch implementation of Stylegan2 with UNet Discriminator

Primary LanguagePythonMIT LicenseMIT

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}
}