search

Zhendong-Wang / Patch-Diffusion

Apache License 2.0
63 stars 11 forks source link

dataset label #4

Open yongliuy opened 7 months ago

yongliuy commented 7 months ago

Hi~

I found in dataset_tool.py that the labels of the training data are all read from dataset.json, but I did not find the corresponding dataset.json in the ImageNet dataset. Can you provide the corresponding json file of ImageNet dataset?

Thanks

Zhendong-Wang commented 7 months ago

You need to preprocess the ImageNet dataset here https://github.com/Zhendong-Wang/Patch-Diffusion?tab=readme-ov-file#preparing-datasets, which will show you the dataset.json in the output zip.