From 4a4c23a66e906f2e82c57464195697d8070805f2 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 25 Jul 2019 13:01:34 +0200 Subject: [PATCH] Discard all images without gt again Signed-off-by: Jim Martens --- src/twomartens/masterthesis/data.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/twomartens/masterthesis/data.py b/src/twomartens/masterthesis/data.py index cf27940..1742c0f 100644 --- a/src/twomartens/masterthesis/data.py +++ b/src/twomartens/masterthesis/data.py @@ -513,7 +513,7 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]], transformations=transformations, label_encoder=label_encoder, returns=returns, - keep_images_without_gt=evaluation + keep_images_without_gt=False ) if debug: @@ -523,7 +523,7 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]], transformations=transformations, label_encoder=label_encoder, returns=returns_debug, - keep_images_without_gt=evaluation + keep_images_without_gt=False ) else: debug_generator = None