From ee5ae5eec028ab237e52c16a677712c3f00d95f9 Mon Sep 17 00:00:00 2001
From: Anton Rigner <arigner@fb.com>
Date: Mon, 12 Jun 2023 15:26:47 -0700
Subject: [PATCH] Draft diff to unblock eval metrics + Eval data loader issue

Differential Revision: D46641931

fbshipit-source-id: f9f9566a646d9bf67e9faeec5d03ab11e22e887e
---
 d2go/data/datasets.py      | 3 +++
 d2go/data/extended_coco.py | 1 +
 2 files changed, 4 insertions(+)

diff --git a/d2go/data/datasets.py b/d2go/data/datasets.py
index 64990448..0fef589e 100644
--- 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
index 0fa8ea7b..e1dc691c 100644
--- 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}