Merge pull request #781 from MouseLand/SimonBon-min_size_fix

Simon bon min size fix
MouseLand · Sep 13, 2023 · a097ecc · a097ecc
2 parents 9f1c6e1 + da6201c
commit a097ecc
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,
+                                                         flow_threshold=flow_threshold, interp=interp, resize=resize, 
+                                                         min_size=min_size if stitch_threshold==0 or nimg==1 else -1, # turn off for 3D stitching
                                                          use_gpu=self.gpu, device=self.device)
                     masks.append(outputs[0])
                     p.append(outputs[1])