Fix matrix dimensions

fantine · fantine · commit fdee4d76cc75 · 2021-05-25T14:59:46.000+02:00
diff --git a/preprocessing/process.py b/preprocessing/process.py
@@ -13,7 +13,7 @@
 
 def _process(data, clip_value, std_channels):
   data = np.clip(data, -clip_value, clip_value)
-  data = data / std_channels
+  data = data / np.expand_dims(std_channels, axis=1)
   return data
 
 
