search

YannDubs / disentangling-vae

Experiments for understanding disentanglement in VAE latent representations
Other
785 stars 143 forks source link

How can I apply it to my own dataset？ #73

Closed hxhebit closed 1 year ago

hxhebit commented 1 year ago

The dataset structure：

|- class 1 | |- class a | | |- item 1 | | |- item 2 | | |..... | |- class b | |- class c | |...... | - class 2 | ......

YannDubs commented 1 year ago

you have to use a pytorch dataloader. Example: https://github.com/YannDubs/disentangling-vae/blob/master/utils/datasets.py#L213 to see what you have to return