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

Remove data that cannot be pickled

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-04-20 11:19:37 +02:00
parent 4ab5ba7745
commit 93c2d88dd4
1 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -266,10 +266,6 @@ def prepare_scenenet_val(data_path: str, protobuf_path: str) -> Tuple[List[List[
instance_dict['coco_id'] = definitions.WNID_TO_COCO[wnid] instance_dict['coco_id'] = definitions.WNID_TO_COCO[wnid]
else: else:
continue # only save instances that are positive instances and not background continue # only save instances that are positive instances and not background
if instance_type == scenenet_pb2.Instance.LIGHT_OBJECT:
instance_dict['light_info'] = instance.light_info
if instance_type == scenenet_pb2.Instance.RANDOM_OBJECT:
instance_dict['object_info'] = instance.object_info
instances_traj_dict[instance_id] = instance_dict instances_traj_dict[instance_id] = instance_dict