Changed code to keep images without GT during evaluation

Signed-off-by: Jim Martens <github@2martens.de>
2019-07-25 12:04:47 +02:00
parent 84adbe21fe
commit 60a9d63a1d
1 changed files with 2 additions and 2 deletions
--- 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