DragGAN icon indicating copy to clipboard operation
DragGAN copied to clipboard

Support for other models used in Original Repo

Open ds-nat-j opened this issue 1 year ago • 1 comments

I've seen other stylegan2 models used in the original demo like stylegan2_human_v2_512.pkl, stylegan2-old-car-config-f.pt etc. If we could parse the model path for these newer models that might come in the future, that would be really awesome.

It would also be great to know how to load these newer models in the code for the time being. Thanks!

ds-nat-j avatar Jun 09 '23 09:06 ds-nat-j

Also I think it's worth investing time adding code that converts the original tensorflow based stylegan2 pre-trained models from .pkl files to PyTorch .pt files within this repo itself.

This can be done using https://github.com/rosinality/stylegan2-pytorch

ds-nat-j avatar Jun 09 '23 18:06 ds-nat-j