stylegan2-pytorch icon indicating copy to clipboard operation
stylegan2-pytorch copied to clipboard

Multiclass training and inference

Open virilo opened this issue 4 years ago • 1 comments

I'd like to stylegan2-pytorch for a multiclass dataset.

The idea is to train the GAN with multiclass.

And during the inference, ask the GAN for an image with multiples tags. I.e. generate an image tagged as 'boat', 'sunset' and 'people'

Is it possible with stylegan2-pytorch?

virilo avatar Jan 20 '21 19:01 virilo

@virilo not at the moment unfortunately, there is actually another issue open for that https://github.com/lucidrains/stylegan2-pytorch/issues/194

lucidrains avatar Jan 20 '21 19:01 lucidrains