Merge branch 'min_size_fix' of https://github.com/SimonBon/cellpose i…

…nto SimonBon-min_size_fix
MouseLand · Sep 13, 2023 · e89c20e · e89c20e
2 parents 9f1c6e1 + e6e19c7
commit e89c20e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cellpose/models.py b/cellpose/models.py
@@ -654,7 +654,7 @@ 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,
                                                          use_gpu=self.gpu, device=self.device)
                     masks.append(outputs[0])
                     p.append(outputs[1])