Skip to content

Commit 432a8c0

Browse files
fix(connected_components_stack): correction for N when bg is present
1 parent 4146715 commit 432a8c0

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

cc3d/__init__.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -483,8 +483,12 @@ def connected_components_stack(
483483
full_binary, mapping = crackle.renumber(full_binary, start=start)
484484
arr = crackle.CrackleArray(full_binary).refit()
485485

486+
N = arr.num_labels()
487+
if start == 0:
488+
N -= 1
489+
486490
if return_N:
487-
return arr, arr.num_labels()
491+
return arr, N
488492
else:
489493
return arr
490494

0 commit comments

Comments
 (0)