Ensure single truth-value.

PiperOrigin-RevId: 588014819
google-research · Dec 6, 2023 · 6bd2cc5 · 6bd2cc5
1 parent 08ca868
commit 6bd2cc5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/connectomics/volume/decorators.py b/connectomics/volume/decorators.py
@@ -392,7 +392,7 @@ def _active_contours_mask(
       _cast_img(data, 'float64').squeeze(),
       num_iter=num_iter_chan_vese).astype(bool)
   if background is not None:
-    mask = ~mask if mask[background] else mask
+    mask = ~mask if mask[background].all() else mask
   if num_iter_erosion > 0:
     mask = scipy.ndimage.binary_erosion(mask, iterations=num_iter_erosion)
   if num_iter_dilation > 0: