Fixed MinMax normalization

VMarsocci · Oct 28, 2024 · 4f318f9 · 4f318f9
1 parent 3fdd72b
commit 4f318f9
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/pangaea/engine/data_preprocessor.py b/pangaea/engine/data_preprocessor.py
@@ -381,9 +381,8 @@ def __call__(
         """
 
         for k in self.data_min.keys():
-            size = (-1,) + data["image"][k].shape[1:]
-            data["image"][k].sub_(self.data_min[k].view(size)).div_(
-                (self.data_max[k] - self.data_min[k]).view(size)
+            data["image"][k].sub_(self.data_min[k].view(-1, 1, 1, 1)).div_(
+                (self.data_max[k] - self.data_min[k]).view(-1, 1, 1, 1)
             )
         return data