A Pytorch implementation of Stylegan2 with UNet Discriminator
MIT License
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.
$ pip install unet-stylegan2
$ unet_stylegan2 --data ./path/to/data
@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}
}