From 60a9d63a1dbd59a74ae2454c2d46ffae64556437 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 25 Jul 2019 12:04:47 +0200 Subject: [PATCH] Changed code to keep images without GT during evaluation 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 c00b714..e1c1f48 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=False + keep_images_without_gt=evaluation ) 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=False + keep_images_without_gt=evaluation ) else: debug_generator = None