search

onnx / models

A collection of pre-trained, state-of-the-art models in the ONNX format
http://onnx.ai/models/
Apache License 2.0
7.71k stars 1.38k forks source link

StyleGAN2 onnx model #462

Open alessiabertugli opened 3 years ago

alessiabertugli commented 3 years ago

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 commented 3 years ago

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

alessiabertugli commented 2 years ago

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

magicse commented 1 year ago

@alessiabertugli GFPGAN to ONNX