From a7a1606ca0dfe52aaaff5b56dd012992587b22b1 Mon Sep 17 00:00:00 2001
From: carsen <carsen.stringer@gmail.com>
Date: Sat, 1 May 2021 20:50:32 -0400
Subject: [PATCH] fixing bug with diameter return

---
 cellpose/models.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/cellpose/models.py b/cellpose/models.py
index 3457b0a1..8ca72483 100644
--- 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,