fixing bug with diameter return

MouseLand · May 2, 2021 · a7a1606 · a7a1606
1 parent 8b52f24
commit a7a1606
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/cellpose/models.py b/cellpose/models.py
@@ -243,6 +243,8 @@ def eval(self, x, batch_size=8, channels=None, channel_axis=None, z_axis=None,
                 reason = 'does not work on non-2D images'
             models_logger.warning(f'could not estimate diameter, {reason}')
             diams = self.diam_mean 
+        else:
+            diams = diameter
 
         tic = time.time()
         masks, flows, styles = self.cp.eval(x,