Processed dataset for the notebook demo

LabForComputationalVision / memorization_generalization_in_diffusion_models

MIT License

30 stars 5 forks source link

Processed dataset for the notebook demo #3

Open masud99r opened 4 months ago

masud99r commented 4 months ago

Thank you for the great work! I am trying to run the notebooks demo notebooks/Demo_UNet_CelebA80x80.ipynb. However, this demo seems to require a processed image dataset "train80x80_no_repeats.pt". Is there any script available to generate them or do you have the processed image that can be downloaded?

Zahra-Kadkhodaie commented 4 months ago

I just added a script, code/celebA_to_torch.py , to pre-process celebA images to 80x80 images we used in our work. To use this you will need to download the celebA dataset, which contains around 202K images. The script then crops and downsamples the images and removes the repeated ones.