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

Added check to prevent frames without labels

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-04-28 18:34:28 +02:00
parent 5fd8ba1ae8
commit 2276d359d6
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -257,8 +257,9 @@ def load_scenenet_val(photo_paths: Sequence[Sequence[str]],
for trajectory in trajectories: for trajectory in trajectories:
traj_image_paths, traj_instances = trajectory traj_image_paths, traj_instances = trajectory
for image_path, frame_instances in zip(traj_image_paths, traj_instances): for image_path, frame_instances in zip(traj_image_paths, traj_instances):
final_image_paths.append(image_path)
labels = [] labels = []
if not frame_instances: # skip frames without instances
continue
for instance in frame_instances: for instance in frame_instances:
bbox = instance['bbox'] bbox = instance['bbox']
labels.append(( labels.append((
@ -269,6 +270,7 @@ def load_scenenet_val(photo_paths: Sequence[Sequence[str]],
bbox[3] bbox[3]
)) ))
final_image_paths.append(image_path)
final_labels.append(labels) final_labels.append(labels)
length_dataset = len(final_image_paths) length_dataset = len(final_image_paths)