diff --git a/scripts/calc_gt_coco.py b/scripts/calc_gt_coco.py index 1667a6f2..1c090af0 100644 --- a/scripts/calc_gt_coco.py +++ b/scripts/calc_gt_coco.py @@ -94,7 +94,7 @@ coco_gt_path = coco_gt_path.replace("scene_gt_coco", "scene_gt_coco_modal") misc.log( "Calculating Coco Annotations - dataset: {} ({}, {}), scene: {}".format( - p["dataset"], p["dataset_split"], p["dataset_split_type"], scene_id + dataset_name, split, split_type, scene_id ) ) diff --git a/scripts/create_pose_results_file_from_gt.py b/scripts/create_pose_results_file_from_gt.py index 6a25ae4a..23290571 100644 --- a/scripts/create_pose_results_file_from_gt.py +++ b/scripts/create_pose_results_file_from_gt.py @@ -19,7 +19,7 @@ # Out perfect result file name "results_name": 'gt-results', # Predefined test targets - "targets_filename": "test_targets_bop24.json", + "targets_filename": "test_targets_bop19.json", # Folder with results to be evaluated. "results_path": config.results_path, # Folder containing the BOP datasets. @@ -88,7 +88,7 @@ target = inout.get_im_targets(img_gt, img_gt_info, p["visib_gt_min"], p["eval_mode"]) for obj_gt in img_gt: - if obj_gt["obj_id"] in target: + if obj_gt["obj_id"] == target["obj_id"]: result = { "scene_id": int(scene_id), "im_id": int(im_id), @@ -98,9 +98,9 @@ "t": obj_gt["cam_t_m2c"], "time": -1.0, } - results.append(result) + results.append(result) result_filename = "{}_{}-{}_pose.csv".format(p["results_name"], p["dataset"], p["split"]) results_path = os.path.join(p["results_path"], result_filename) inout.save_bop_results(results_path, results) -print('Saved ', results_path) +print('Saved ', results_path) \ No newline at end of file