fix inverted EDT processing for dask arrays / non-binary masks (#146)

royerlab · Oct 7, 2024 · 23e41db · 23e41db
1 parent 3e29caa
commit 23e41db
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ultrack/imgproc/segmentation.py b/ultrack/imgproc/segmentation.py
@@ -204,6 +204,7 @@ def inverted_edt(
     ArrayLike
         Inverted and normalized EDT.
     """
+    mask = np.asarray(mask)
     if axis is None:
         dist = edt.edt(mask, anisotropy=voxel_size)
     else:
@@ -216,7 +217,7 @@ def inverted_edt(
         )
     dist = dist / dist.max()
     dist = 1.0 - dist
-    dist[~mask] = 1
+    dist[mask == 0] = 1
     return dist