fix generic tensorflow preprocessing

ViCCo-Group · Nov 26, 2024 · 1f95a73 · 1f95a73
1 parent d606b8f
commit 1f95a73
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/thingsvision/core/extraction/tensorflow.py b/thingsvision/core/extraction/tensorflow.py
@@ -88,7 +88,7 @@ def get_default_transformation(
         apply_center_crop: bool = True,
     ) -> Any:
         resize_dim = crop_dim
-        composes = [layers.experimental.preprocessing.Resizing(resize_dim, resize_dim)]
+        composes = [layers.experimental.preprocessing.Resizing(resize_dim, resize_dim), layers.experimental.preprocessing.Rescaling(1./255.)]
         if apply_center_crop:
             pass
             # TODO: fix center crop problem with Keras