From 5f9e8d52f30a05d07ecc5cb44f818feac35cb19d Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Fri, 19 Apr 2019 15:09:46 +0200 Subject: [PATCH] Fixed wrong attribute usage Signed-off-by: Jim Martens --- src/twomartens/masterthesis/data.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/data.py b/src/twomartens/masterthesis/data.py index d42c0a2..ee881f4 100644 --- a/src/twomartens/masterthesis/data.py +++ b/src/twomartens/masterthesis/data.py @@ -266,7 +266,7 @@ def prepare_scenenet_val(data_path: str, protobuf_path: str) -> Tuple[List[List[ else: instance_dict['coco_id'] = 0 # if no COCO id is found, the correct COCO class is background if instance_type == scenenet_pb2.Instance.LIGHT_OBJECT: - instance_dict['light_type'] = instance.light_type + instance_dict['light_info'] = instance.light_info if instance_type == scenenet_pb2.Instance.RANDOM_OBJECT: instance_dict['object_info'] = instance.object_info