StyleGAN2 onnx model

[alessiabertugli]

Model Request

Which model would you like to see in the model zoo?

StyleGAN2 generator on faces (CelebA, FFHQ, ...) Paper: "Analyzing and Improving the Image Quality of StyleGAN", https://arxiv.org/abs/1912.04958

Describe why this model is relevant

The environment to make inferences on this model is complicated to set up. Moreover, it works only on old Cuda versions and old GPUs.

Model information

Source repository: https://github.com/NVlabs/stylegan2.git Model Zoo category: Vision Model Zoo sub-category: Image Generation

[wenbingl]

this looks tensorflow model, can you try this tool https://github.com/onnx/tensorflow-onnx to convert it to ONNX model?

[alessiabertugli]

Sorry, I share you the pytorch version https://github.com/rosinality/stylegan2-pytorch

[magicse]

@alessiabertugli GFPGAN to ONNX