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

Changed code to keep images without GT during evaluation

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-25 12:04:47 +02:00
parent 84adbe21fe
commit 60a9d63a1d
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=False keep_images_without_gt=evaluation
) )
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=False keep_images_without_gt=evaluation
) )
else: else:
debug_generator = None debug_generator = None