Discard all images without gt again

Signed-off-by: Jim Martens <github@2martens.de>
2019-07-25 13:01:34 +02:00
parent e213ed62ec
commit 4a4c23a66e
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=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