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

Added inverse transforms to output of non-training generators

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-13 16:06:21 +02:00
parent cc74a8674e
commit f8bed423e4
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -303,6 +303,11 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]],
width=resized_shape[1]) width=resized_shape[1])
] ]
returns = {'processed_images', 'encoded_labels'}
if not training:
returns.update({'inverse_transform'})
generator = data_generator.generate( generator = data_generator.generate(
batch_size=batch_size, batch_size=batch_size,
shuffle=shuffle, shuffle=shuffle,
@ -320,7 +325,7 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]],
[1.0, 2.0, 0.5], [1.0, 2.0, 0.5],
[1.0, 2.0, 0.5]] [1.0, 2.0, 0.5]]
), ),
returns={'processed_images', 'encoded_labels'}, returns=returns,
keep_images_without_gt=False keep_images_without_gt=False
) )