apply nomalization to rgb data along the time dim

aazuspan · Dec 13, 2024 · bac6509 · bac6509
1 parent 1c41593
commit bac6509
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/wxee/xarray.py b/wxee/xarray.py
@@ -130,7 +130,7 @@ def normalize(self, stretch: float = 1.0) -> xr.DataArray:
         if stretch < 0 or stretch > 1:
             raise ValueError("Stretch value must be in the range [0.0, 1.0].")
 
-        min_val = da.quantile(1 - stretch)
-        max_val = da.quantile(stretch)
+        min_val = da.quantile(1 - stretch, dim=["x", "y", "variable"])
+        max_val = da.quantile(stretch, dim=["x", "y", "variable"])
 
         return ((da - min_val) / (max_val - min_val)).clip(0, 1)