From 8bddf320f8f2950f564d073a7e2e9ab89619372d Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Tue, 2 Jul 2019 11:51:23 +0200 Subject: [PATCH] Added wordnet class name to prepared ground truth Signed-off-by: Jim Martens --- src/twomartens/masterthesis/data.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/twomartens/masterthesis/data.py b/src/twomartens/masterthesis/data.py index 952eb09..930c7a7 100644 --- a/src/twomartens/masterthesis/data.py +++ b/src/twomartens/masterthesis/data.py @@ -435,7 +435,9 @@ def prepare_scenenet_data(data_path: str, protobuf_path: str) -> Tuple[List[List instance_dict = {} if instance_type != scenenet_pb2.Instance.BACKGROUND: wnid = instance.semantic_wordnet_id + wn_class = instance.semantic_english instance_dict['wordnet_id'] = wnid + instance_dict['wordnet_class_name'] = wn_class if wnid in definitions.WNID_TO_COCO: instance_dict['coco_id'] = definitions.WNID_TO_COCO[wnid] else: