zhiCHEN96 / ConceptWhitening

MIT License
118 stars 28 forks source link

Regarding the concept datasets #2

Closed sakura7621 closed 3 years ago

sakura7621 commented 3 years ago

Hi,

I have some problems with the concept datasets while trying to re-train the model. The data from MS COCO (https://cocodataset.org) contains images that need to be pre-processed (e.g., cropped) before feeding into the model. Can you please provide the concept datasets?

Thank you!

zhiCHEN96 commented 3 years ago

Hi,

We just upload the preprocessing code for the COCO dataset. Please check cropping_images_COCO.py.