From ee74480dd76f157b36329d3471ef1c0d1ec3e7fb Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Sun, 28 Apr 2019 18:38:27 +0200 Subject: [PATCH] Optimized check for empty labels Signed-off-by: Jim Martens --- src/twomartens/masterthesis/data.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/twomartens/masterthesis/data.py b/src/twomartens/masterthesis/data.py index c955386..a9881f4 100644 --- a/src/twomartens/masterthesis/data.py +++ b/src/twomartens/masterthesis/data.py @@ -258,8 +258,6 @@ def load_scenenet_val(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: # skip frames without instances - continue for instance in frame_instances: bbox = instance['bbox'] labels.append(( @@ -269,6 +267,9 @@ def load_scenenet_val(photo_paths: Sequence[Sequence[str]], bbox[2], bbox[3] )) + + if not labels: + continue final_image_paths.append(image_path) final_labels.append(labels) @@ -276,7 +277,7 @@ def load_scenenet_val(photo_paths: Sequence[Sequence[str]], length_dataset = len(final_image_paths) path_dataset = tf.data.Dataset.from_tensor_slices(final_image_paths) - label_dataset = tf.data.Dataset.from_sparse_tensor_slices(final_labels) + label_dataset = tf.data.Dataset.from_tensor_slices(final_labels) dataset = tf.data.Dataset.zip((path_dataset, label_dataset)) dataset = dataset.apply(tf.data.experimental.shuffle_and_repeat(buffer_size=length_dataset, count=num_epochs)) dataset = dataset.batch(batch_size=batch_size)