in scripts/data_preparation/prepare_vistas.py [0:0]
def _cat_id_maps(categories):
cat_id_mvd_to_iss = dict()
cat_id_iss_to_mvd = dict()
num_thing, num_stuff = 0, 0
# Find stuff
for cat_id, cat_meta in enumerate(categories):
if not cat_meta["evaluate"]:
continue
if not cat_meta["instances"]:
cat_id_mvd_to_iss[cat_id] = num_stuff
cat_id_iss_to_mvd[num_stuff] = cat_id
num_stuff += 1
for cat_id, cat_meta in enumerate(categories):
if not cat_meta["evaluate"]:
continue
if cat_meta["instances"]:
cat_id_mvd_to_iss[cat_id] = num_thing + num_stuff
cat_id_iss_to_mvd[num_thing + num_stuff] = cat_id
num_thing += 1
return cat_id_mvd_to_iss, cat_id_iss_to_mvd, num_stuff, num_thing