How do I build a image dataset with for IDDPM which uses labels for training?

open-mmlab / mmgeneration

MMGeneration is a powerful toolkit for generative models, based on PyTorch and MMCV.

https://mmgeneration.readthedocs.io/en/latest/

Apache License 2.0

1.91k stars 232 forks source link

How do I build a image dataset with for IDDPM which uses labels for training? #479

Open CreamyLong opened 2 years ago

CreamyLong commented 2 years ago

I find that there is no dataset script in mmgen/datasets for conditional IDDPM traning.

LeoXing1996 commented 2 years ago

Hey @CreamyLong, we have not support conditional training for IDDPM currently. If you want to a conditional one by yourself, I think you can use CIFAR10 and ImageNet.