From a387cfb47740bb6408c9eb3eb30a4d8d6a152e3a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Al=C3=A1n=20Mu=C3=B1oz?= <amuoz@ed.ac.uk>
Date: Tue, 18 Jan 2022 18:16:36 +0000
Subject: [PATCH] bugfix
---
extraction/core/functions/cell.py | 10 ++++++++--
1 file changed, 8 insertions(+), 2 deletions(-)
diff --git a/extraction/core/functions/cell.py b/extraction/core/functions/cell.py
index 77179cc4..cd52aeb3 100644
--- a/extraction/core/functions/cell.py
+++ b/extraction/core/functions/cell.py
@@ -54,7 +54,10 @@ def max5px_med(cell_mask, trap_image):
top_vals = sorted_vals[-5:]
max5px = np.mean(top_vals, dtype=float)
- return max5px / sorted_vals[len(sorted_vals) // 2]
+ if max5px:
+ return max5px / sorted_vals[len(sorted_vals) // 2]
+ else:
+ return max2p5px
def max2p5pc_med(cell_mask, trap_image):
@@ -65,7 +68,10 @@ def max2p5pc_med(cell_mask, trap_image):
top_vals = sorted_vals[-top_pixels:]
max2p5pc = np.mean(top_vals, dtype=float)
- return max2p5pc / sorted_vals[len(sorted_vals) // 2]
+ if max2p5pc:
+ return max2p5pc / sorted_vals[len(sorted_vals) // 2]
+ else:
+ return max2p5pc
def std(cell_mask, trap_image):
--
GitLab