Remove images from data set that do not have instances

Signed-off-by: Jim Martens <github@2martens.de>
2019-06-18 10:25:59 +02:00
parent 9a5c96c46c
commit 9153828ff5
1 changed files with 4 additions and 0 deletions
--- a/src/twomartens/masterthesis/data.py
+++ b/src/twomartens/masterthesis/data.py
@ -269,6 +269,10 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]],
        traj_image_paths, traj_instances = trajectory
        for image_path, frame_instances in zip(traj_image_paths, traj_instances):
            labels = []
+            
+            if not frame_instances:
+                continue  # skip images that do not contain instances
+            
            for instance in frame_instances:
                bbox = instance['bbox']
                labels.append([