Draft diff to unblock eval metrics + Eval data loader issue

Differential Revision: D46641931 fbshipit-source-id: f9f9566a646d9bf67e9faeec5d03ab11e22e887e
dxzhou2023 · Jun 12, 2023 · ee5ae5e · ee5ae5e
1 parent bcad53f
commit ee5ae5e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/d2go/data/datasets.py b/d2go/data/datasets.py
@@ -55,6 +55,9 @@ def _register_extended_coco(dataset_name, split_dict):
  json_file=json_file,
  image_root=image_root,
  dataset_name=dataset_name,
+ image_direct_copy_keys=[
+ "inst_segm_file_name",
+ ],
  **load_kwargs,
  )
  DatasetCatalog.register(dataset_name, load_coco_json_func)

diff --git a/d2go/data/extended_coco.py b/d2go/data/extended_coco.py
@@ -180,6 +180,7 @@ def convert_coco_annotations(
  "point_labels",
  "associations",
  "file_name",
+ "inst_segm_id",
  ]
  # NOTE: maybe use MetadataCatalog for this
  obj = {field: anno[field] for field in fields_to_copy if field in anno}