dome272 / Diffusion-Models-pytorch

Pytorch implementation of Diffusion Models (https://arxiv.org/pdf/2006.11239.pdf)
Apache License 2.0
1.11k stars 256 forks source link

dataset #12

Open InsaneGe opened 1 year ago

InsaneGe commented 1 year ago

First, thank you for useful video and code. can you share the dataset that you use in the code?I want to apply the code in cifar-10, but failed, so I want to have a look at your dateset structure used in the code.

args.dataset_path = r"C:\Users\dome\datasets\landscape_img_folder" the complete file of "landscape_img_folder"

dome272 commented 1 year ago

Do you need the landscape dataset or the cifar-10? The landscape dataset can be found here: https://www.kaggle.com/datasets/arnaud58/landscape-pictures and the CIFAR-10 here: https://www.kaggle.com/datasets/joaopauloschuler/cifar10-64x64-resized-via-cai-super-resolution

And the structure is simply: landscape_dataset/unconditional/{0..6000}.jpg and for cifar 10 you would have cifar_10/ and then 10 subfolders for each class and then all images in all of the folders for each class

potatoker commented 1 year ago

@InsaneGe got same error here, you need to unzip the landscape dataset to the specified directory named "unconditional" which should created by hand.