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

Fixed wrong attribute usage

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-04-19 15:09:46 +02:00
parent 37e355e820
commit 5f9e8d52f3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -266,7 +266,7 @@ def prepare_scenenet_val(data_path: str, protobuf_path: str) -> Tuple[List[List[
else: else:
instance_dict['coco_id'] = 0 # if no COCO id is found, the correct COCO class is background 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: 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: if instance_type == scenenet_pb2.Instance.RANDOM_OBJECT:
instance_dict['object_info'] = instance.object_info instance_dict['object_info'] = instance.object_info