MAINT: Allow 'None' segmentation to turn off segmentations.

pmgbergen · Jun 12, 2024 · 1e993aa · 1e993aa
1 parent 7548712
commit 1e993aa
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/darsia/presets/fluidflower/simplefluidflower.py b/src/darsia/presets/fluidflower/simplefluidflower.py
@@ -129,7 +129,10 @@ def setup(
             self.baseline = correction(self.baseline)
 
         # ! ---- SETUP SEGMENTATION ----
-        self.labels = self.setup_segmentation(segmentation)
+        if segmentation is None:
+            self.labels = darsia.zeros_like(self.baseline, dtype=np.uint8)
+        else:
+            self.labels = self.setup_segmentation(segmentation)
 
         # ! ---- SETUP COLOR CORRECTION ----
         illumination_mode = kwargs.get("illumination_mode", "automatic")