stylegan-human/StyleGAN-Human

[QUESTION] Use custom dataset lables

Avedena opened this issue · 1 comments

How I can Prepare my Dataset with Labels to Train this Model?
I lable my data set as provided here in the PyTorch Docs

Here a Example frome my CSV:
"Image_Name""test.png", "Male":True, "Hair_Black":False, "Hair_Brown":True

The original StyleGAN Lib use only Integer Lables. How I can get the benefits in Learning from my Dataset?
Thanks

Hi @Avedena

you can use some pre-defined integer labels to replace your original labels. For example, use 0 to denote "Female: True", use 1 to denote "Male: True".

Thanks!