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()