2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Added wordnet class name to prepared ground truth

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-02 11:51:23 +02:00
parent 3d7d97355c
commit 8bddf320f8
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/twomartens/masterthesis/data.py
View File

@ -435,7 +435,9 @@ def prepare_scenenet_data(data_path: str, protobuf_path: str) -> Tuple[List[List
instance_dict = {} instance_dict = {}
if instance_type != scenenet_pb2.Instance.BACKGROUND: if instance_type != scenenet_pb2.Instance.BACKGROUND:
wnid = instance.semantic_wordnet_id wnid = instance.semantic_wordnet_id
wn_class = instance.semantic_english
instance_dict['wordnet_id'] = wnid instance_dict['wordnet_id'] = wnid
instance_dict['wordnet_class_name'] = wn_class
if wnid in definitions.WNID_TO_COCO: if wnid in definitions.WNID_TO_COCO:
instance_dict['coco_id'] = definitions.WNID_TO_COCO[wnid] instance_dict['coco_id'] = definitions.WNID_TO_COCO[wnid]
else: else: