z-axis=0 in GUI (#1020)

MouseLand · Oct 25, 2024 · 6fb288f · 6fb288f
1 parent 166abf1
commit 6fb288f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cellpose/gui/gui.py b/cellpose/gui/gui.py
@@ -2431,7 +2431,7 @@ def compute_segmentation(self, custom=False, model_name=None, load_model=True):
                     cellprob_threshold=cellprob_threshold,
                     flow_threshold=flow_threshold, do_3D=do_3D, niter=niter,
                     normalize=normalize_params, stitch_threshold=stitch_threshold,
-                    progress=self.progress)[:2]
+                    progress=self.progress, z_axis=0 if self.NZ > 1 else None)[:2]
             except Exception as e:
                 print("NET ERROR: %s" % e)
                 self.progress.setValue(0)