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

Fixed comparison of coordinates

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-04-28 18:19:28 +02:00
parent ad11a5f6c6
commit d7ef9513a7
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

@ -395,7 +395,7 @@ def prepare_scenenet_val(data_path: str, protobuf_path: str) -> Tuple[List[List[
instance_local[instance_local != instance_id] = 0 instance_local[instance_local != instance_id] = 0
instance_local[instance_local == instance_id] = 1 instance_local[instance_local == instance_id] = 1
coordinates = ndimage.find_objects(instance_local) coordinates = ndimage.find_objects(instance_local)
if not coordinates: # the current instance was not in this frame if coordinates is None: # the current instance was not in this frame
continue continue
else: else:
coordinates = coordinates[0] # extract the coords of the one object coordinates = coordinates[0] # extract the coords of the one object