search

zhiCHEN96 / ConceptWhitening

MIT License
118 stars 28 forks source link

Running Module #1

Open AndrewSDennisRepo opened 3 years ago

AndrewSDennisRepo commented 3 years ago

Hello,

I hope you are well. I am trying to run this module and not having full success. I trained the model on places365 and it seems to have only saved 1 checkpoint so there is a loading error when trying to evaluate.

Also, what is the best link to get your exact concept_whitening training? COCO supplies just a folder of photos do we need to use the annotations to create the airplane_bed_person. I would like to use the exact same data and potentially just use your weights to test this out.

Thank you!

zhiCHEN96 commented 3 years ago

Hi,

Would you provide more information about the loading error?

Also, in terms of dealing with the COCO dataset, we just upload the preprocessing code. Please check cropping_images_COCO.py.