def __init__(self, run_path, model_ckpt):
self.lvis_gt = LVIS(ANNOTATION_PATH)
self.lvis_dt = LVISResults(self.lvis_gt, PREDICTION_PATH)
self.run_path = run_path
self.model_ckpt = model_ckpt
self._build_coco_to_lvis_map()
cocoEval = LVISEval(self.lvis_gt, self.lvis_dt, 'segm')
self.freq_groups = cocoEval._prepare_freq_group()
config_path = os.path.join(self.run_path, 'config_lvis.yaml')
self.config = yaml.load(open(config_path, "r"), Loader=yaml.FullLoader)
def __init__(self):
self.lvis = LVIS('/scratch/users/zzweng/datasets/lvis/lvis_v0.5_val.json')
self.dt_path = 'output/inference/lvis_instances_results.json'
self.lvis_dt = LVISResults(self.lvis, self.dt_path)
coco_map = json.load(open('lvis-api/data/coco_to_synset.json'))
synset_to_lvis = {cat['synset']: cat['id'] for cat in self.lvis.cats.values()}
synset_to_lvis['oven.n.01'] = synset_to_lvis['toaster_oven.n.01']
synset_to_lvis['frank.n.02'] = synset_to_lvis['sausage.n.01']
coco_to_lvis = {}
lvis_to_coco = {}
for item in coco_map.values():
coco_id, lvis_id = item['coco_cat_id'], synset_to_lvis[item['synset']]
coco_to_lvis[coco_id] = lvis_id
lvis_to_coco[lvis_id] = coco_id
self.coco_to_lvis = coco_to_lvis
self.lvis_to_coco = lvis_to_coco
cocoEval = LVISEval(self.lvis, self.lvis_dt,'segm')
self.freq_groups = cocoEval._prepare_freq_group()
