how to use condition on CelebA data?

andreas128 / RePaint

Official PyTorch Code and Models of "RePaint: Inpainting using Denoising Diffusion Probabilistic Models", CVPR 2022

1.88k stars 153 forks source link

how to use condition on CelebA data? #13

Open gozi1123 opened 1 year ago

gozi1123 commented 1 year ago

I want to use the celebA ckpt (with masks), but use a condition on the mask (like the classifier for the imagenet ckpt). for example, use the desired mean pixel value (of the mask) as condition for the model. How can I use it? Can I use my condition as input for the model?