diff --git a/cellpose/models.py b/cellpose/models.py index cbb1f3c6..6fb8591f 100644 --- 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])