adding if statement for 3D stitching

MouseLand · Sep 13, 2023 · 958e0a6 · 958e0a6
1 parent e89c20e
commit 958e0a6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cellpose/models.py b/cellpose/models.py
@@ -654,7 +654,8 @@ def _run_cp(self, x, compute_masks=True, normalize=True, invert=False,
                 resize = [shape[1], shape[2]] if not resample else None
                 for i in iterator:
                     outputs = dynamics.compute_masks(dP[:,i], cellprob[i], niter=niter, cellprob_threshold=cellprob_threshold,
-                                                         flow_threshold=flow_threshold, interp=interp, resize=resize, min_size=min_size,
+                                                         flow_threshold=flow_threshold, interp=interp, resize=resize, 
+                                                         min_size=min_size if stitch_threshold==0 or nimg==1 else None,
                                                          use_gpu=self.gpu, device=self.device)
                     masks.append(outputs[0])
                     p.append(outputs[1])