EugenHotaj / pytorch-generative

Easy generative modeling in PyTorch
MIT License
428 stars 68 forks source link

How to get the labels of generated images #37

Open Eureka-Y opened 1 year ago

Eureka-Y commented 1 year ago

Hi Eugen, I have successfully generated the image with NADE model,but i don't konw how to generate the label of the image.I don't see relative functions in codes.Can i get the label by some functions?

EugenHotaj commented 1 year ago

@Eureka-Y there is no easy way to get the label of the generated image because the models are not conditioned on the label. There are a few ways you could get labels, but these require significant effort:

  1. Use a discriminative model to predict the label of the generated images.
  2. Modify the models to be conditioned on the label so you can control which class the images are generated from.
  3. Modify the models to also output the label of the generated image (this might not work too well in practice).