betterze / StyleSpace

StyleSpace Analysis: Disentangled Controls for StyleGAN Image Generation
309 stars 34 forks source link

Binary classifier training code for face attributes #18

Closed bwhwang closed 2 years ago

bwhwang commented 2 years ago

I'm really impressed for your great works.

Actually, we would like to add more control attributes for modifying facial image. Instead of applying a pre-trained weights for a binary classifier, we want to train the classifier for a new attribute. Where can we get a training code for the binary classifier based on stylegan discriminator? Thank you in advance.

betterze commented 2 years ago

We take the binary classifiers from StyleGAN in here. We dont train them by ourselves, so we dont have the codes.

But I believe you could just fine tune some face related networks, for example, arcface.