bryandlee / repurpose-gan

Implementation of Repurposing GANs for One-shot Semantic Part Segmentation
126 stars 17 forks source link

Can you use any StyleGAN 2 ADA Generator? #8

Closed MationPlays closed 2 years ago

MationPlays commented 2 years ago

Hey, I want to use the styleGAN2 ADA Pytorch to train with unlabeled data of cars. I want to use the stylegan2 ada pytorch from this Video: https://www.youtube.com/watch?v=BCde68k6KXg&t=0s&ab_channel=JeffHeaton Can there be compatibility problems with Generators from other stylegan versions or is there no difference? I am kind of new to stylegan. Thanks

bryandlee commented 2 years ago

This repo uses rosinality's implementation of stylegan2.

The one you mentioned is an official Nvidia version, and I believe you can convert the model weights: https://github.com/rosinality/stylegan2-pytorch/issues/206

MationPlays commented 2 years ago

This repo uses rosinality's implementation of stylegan2.

The one you mentioned is an official Nvidia version, and I believe you can convert the model weights: rosinality/stylegan2-pytorch#206

Thanks for the reply but I rather stick with the rosinality one so I hope there won't be much problems