How to train the StyleGAN2Generator used as prior knowledge if i want to make some modifications #326

Open
opened 2026-01-29 21:46:51 +00:00 by claunia · 3 comments
Owner

Originally created by @NJUPT-IPR-ZhuWeihong on GitHub (Mar 22, 2023).

Ur work is amazing! I want to make some modifications in the StyleGAN2Generator network.Due to modifications above,the pretrained models of StyleGAN2Generator cannot be used any more.What's more,as ur paper mentioned, generative facial prior gives rise to performance improvement.
Therefore, I have to train StyleGAN2Generator solely to achieve better performance.I am wondering if i could just modify the official Pytorch StyleGAN2 code(the train file, mainly the network architecture) and the model file will adapt to GFPGAN. Of course i modify GFPGAN(the train file and the inference file) at the same time.
Is there any tips or warnings?
It will be appreciated a lot if i could get ur guidance.
Whatever, thanks for ur amazing work in FSR again!

Originally created by @NJUPT-IPR-ZhuWeihong on GitHub (Mar 22, 2023). Ur work is amazing! I want to make some modifications in the StyleGAN2Generator network.Due to modifications above,the pretrained models of StyleGAN2Generator cannot be used any more.What's more,as ur paper mentioned, generative facial prior gives rise to performance improvement. Therefore, I have to train StyleGAN2Generator solely to achieve better performance.I am wondering if i could just modify the official Pytorch StyleGAN2 code(the train file, mainly the network architecture) and the model file will adapt to GFPGAN. Of course i modify GFPGAN(the train file and the inference file) at the same time. Is there any tips or warnings? It will be appreciated a lot if i could get ur guidance. Whatever, thanks for ur amazing work in FSR again!
Author
Owner

@NJUPT-IPR-ZhuWeihong commented on GitHub (Mar 22, 2023):

I found a yml file in basicsr\options\train\StyleGAN, named train_StyleGAN2_256_Cmul2_FFHQ.yml.Maybe it will work?
I don't know.Trying...

@NJUPT-IPR-ZhuWeihong commented on GitHub (Mar 22, 2023): I found a yml file in basicsr\options\train\StyleGAN, named train_StyleGAN2_256_Cmul2_FFHQ.yml.Maybe it will work? I don't know.Trying...
Author
Owner

@zzuyb commented on GitHub (May 17, 2023):

I found a yml file in basicsr\options\train\StyleGAN, named train_StyleGAN2_256_Cmul2_FFHQ.yml.Maybe it will work? I don't know.Trying...

Can it be used normally?

@zzuyb commented on GitHub (May 17, 2023): > I found a yml file in basicsr\options\train\StyleGAN, named train_StyleGAN2_256_Cmul2_FFHQ.yml.Maybe it will work? I don't know.Trying... Can it be used normally?
Author
Owner

@NJUPT-IPR-ZhuWeihong commented on GitHub (May 18, 2023):

I found a yml file in basicsr\options\train\StyleGAN, named train_StyleGAN2_256_Cmul2_FFHQ.yml.Maybe it will work? I don't know.Trying...

Can it be used normally?

yes,just modify some parameters according to ur requirements

@NJUPT-IPR-ZhuWeihong commented on GitHub (May 18, 2023): > > I found a yml file in basicsr\options\train\StyleGAN, named train_StyleGAN2_256_Cmul2_FFHQ.yml.Maybe it will work? I don't know.Trying... > > Can it be used normally? yes,just modify some parameters according to ur requirements
Sign in to join this conversation.
1 Participants
Notifications
Due Date
No due date set.
Dependencies

No dependencies set.

Reference: TencentARC/GFPGAN#326