betterze / StyleSpace

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

About pretrained classifiers #24

Closed sarmientoj24 closed 2 years ago

sarmientoj24 commented 2 years ago
  1. Is the pretrained classifiers for car and LSUN bedroom available?
  2. How do I create or find a pretrained classifier for a model I trained from scratch?
  3. This notebook seems to be very specialized for FFHQ. Any advise or notebook you used for other classes?

Wonderful work!

betterze commented 2 years ago
  1. Here are some pretrained classifiers for bedroom. But human face domain has more anotation data to trained classfiers on.
  2. The classifiers are trained on external data sets. It will be hard to get them in bedroom domain.
  3. Currently, I only create the nootbook for ffhq. You could create simiar notebook for other domains by run localized-channels detection.
sarmientoj24 commented 2 years ago

Thank you for this. However, I do not completely understand which ones are the pretrained classifier for the bedroom in the link.