2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Discard all images without gt again

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-25 13:01:34 +02:00
parent e213ed62ec
commit 4a4c23a66e
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/twomartens/masterthesis/data.py
View File

@ -513,7 +513,7 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]],
transformations=transformations, transformations=transformations,
label_encoder=label_encoder, label_encoder=label_encoder,
returns=returns, returns=returns,
keep_images_without_gt=evaluation keep_images_without_gt=False
) )
if debug: if debug:
@ -523,7 +523,7 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]],
transformations=transformations, transformations=transformations,
label_encoder=label_encoder, label_encoder=label_encoder,
returns=returns_debug, returns=returns_debug,
keep_images_without_gt=evaluation keep_images_without_gt=False
) )
else: else:
debug_generator = None debug_generator = None