threshold applied properly

SchmidtDSE · Jun 5, 2024 · 5a73bf4 · 5a73bf4
1 parent fd90cdf
commit 5a73bf4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/lib/derive_boundary.py b/src/lib/derive_boundary.py
@@ -17,7 +17,7 @@ def apply(self, metric_layer):
 class OtsuThreshold(ThresholdingStrategy):
     def apply(self, metric_values):
         threshold = threshold_otsu(metric_values)
-        burn_mask = metric_values.where(metric_values >= threshold, 1, 0)
+        burn_mask = (metric_values >= threshold).astype(int)
         return burn_mask
 
 
@@ -26,7 +26,7 @@ def __init__(self, threshold=0.5):
         self.threshold = threshold
 
     def apply(self, metric_values):
-        burn_mask = metric_values.where(metric_values >= self.threshold, 1, 0)
+        burn_mask = (metric_values >= self.threshold).astype(int)
         return burn_mask