From 2276d359d69ed772ac9e9528238542d92b7160eb Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Sun, 28 Apr 2019 18:34:28 +0200 Subject: [PATCH] Added check to prevent frames without labels Signed-off-by: Jim Martens --- src/twomartens/masterthesis/data.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/twomartens/masterthesis/data.py b/src/twomartens/masterthesis/data.py index abb2c0e..c955386 100644 --- a/src/twomartens/masterthesis/data.py +++ b/src/twomartens/masterthesis/data.py @@ -257,8 +257,9 @@ def load_scenenet_val(photo_paths: Sequence[Sequence[str]], for trajectory in trajectories: traj_image_paths, traj_instances = trajectory for image_path, frame_instances in zip(traj_image_paths, traj_instances): - final_image_paths.append(image_path) labels = [] + if not frame_instances: # skip frames without instances + continue for instance in frame_instances: bbox = instance['bbox'] labels.append(( @@ -268,7 +269,8 @@ def load_scenenet_val(photo_paths: Sequence[Sequence[str]], bbox[2], bbox[3] )) - + + final_image_paths.append(image_path) final_labels.append(labels) length_dataset = len(final_image_paths)