bug fix yeaz tracker

SchmollerLab · Jan 28, 2022 · f9417a8 · f9417a8
1 parent d54d00d
commit f9417a8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cellacdc/models/YeaZ/unet/tracking.py b/cellacdc/models/YeaZ/unet/tracking.py
@@ -78,7 +78,7 @@ def correspondence_stack(stack, signals=None):
     corrects correspondence of a stack of segmented and labeled masks, by
     fitting the hungarian iteratively on the stack
     """
-    corrected_stack = np.empty(stack.shape)
+    corrected_stack = np.empty(stack.shape, dtype=np.uint16)
     corrected_stack[0] = stack[0]
     for idx in range(len(stack)):
         try: