fix: incorrect result for k=1

seung-lab · Nov 17, 2024 · d48f6c6 · d48f6c6
1 parent b7dc75d
commit d48f6c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cc3d/__init__.py b/cc3d/__init__.py
@@ -114,7 +114,7 @@ def largest_k(
   cts = statistics(cc_labels, no_slice_conversion=True)["voxel_counts"]  
 
   if k == 1:
-    cc_out = (cc_labels == np.argmax(cts))
+    cc_out = (cc_labels == (np.argmax(cts[1:]) + 1))
     if return_N:
       return cc_out, 1
     return cc_out