feat: support binary_image flag in connected_components_stack

seung-lab · Oct 9, 2024 · 4406885 · 4406885
1 parent 24bcc9a
commit 4406885
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/cc3d/__init__.py b/cc3d/__init__.py
@@ -155,6 +155,7 @@ def connected_components_stack(
   connectivity:int = 26,
   return_N:bool = False,
   out_dtype:Optional[Any] = None,
+  binary_image:bool = False,
 ):
   """
   This is for performing connected component labeling
@@ -201,6 +202,7 @@ def connected_components_stack(
     cc_labels, N = connected_components(
       image, connectivity=connectivity,
       return_N=True, out_dtype=np.uint64,
+      binary_image=bool(binary_image),
     )
     cc_labels[cc_labels != 0] += offset
     offset += N