search

princetonvisualai / revise-tool

REVISE: A Tool for Measuring and Mitigating Bias in Visual Datasets --- https://arxiv.org/abs/2004.07999
MIT License
111 stars 19 forks source link

Dataset Reference for COCO #2

Closed drishyamlabs closed 4 years ago

drishyamlabs commented 4 years ago

Hi,

First of all thanks a lot for releasing wonderful tool. Kudos. I am struggling with running the tool properly on COCO dataset as this seems to be a missing link-

gender_data = pickle.load(open('Data/Coco/2014data/bias_splits/train.data', 'rb'))

Also, I am not able to understand what is happening in the next line:

self.gender_info = {int(chunk['img'][15:27]): chunk['annotation'][0] for chunk in gender_data}

Can you please guide me with the path to download this pickle file which is acceptable by the code.

Angelina-Wang commented 4 years ago

yes good point my bad! The gender annotations come from a previous work, Men Also Like Shopping, and those data files are located here!