Added inverse transforms to output of non-training generators

Signed-off-by: Jim Martens <github@2martens.de>

src/twomartens/masterthesis/data.py

@@ -303,6 +303,11 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]],
                                                      width=resized_shape[1])
         ]
         
+    returns = {'processed_images', 'encoded_labels'}
+    
+    if not training:
+        returns.update({'inverse_transform'})
+    
     generator = data_generator.generate(
         batch_size=batch_size,
         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]]
         ),
-        returns={'processed_images', 'encoded_labels'},
+        returns=returns,
         keep_images_without_gt=False
     )