From bff61301419839e7774cb982270213ab33c5d9c7 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:30:04 +0000
Subject: [PATCH] bugfix
---
extraction/core/functions/cell.py | 12 ++++--------
1 file changed, 4 insertions(+), 8 deletions(-)
diff --git a/extraction/core/functions/cell.py b/extraction/core/functions/cell.py
index cd52aeb3..9eea1ced 100644
--- a/extraction/core/functions/cell.py
+++ b/extraction/core/functions/cell.py
@@ -54,10 +54,8 @@ def max5px_med(cell_mask, trap_image):
top_vals = sorted_vals[-5:]
max5px = np.mean(top_vals, dtype=float)
- if max5px:
- return max5px / sorted_vals[len(sorted_vals) // 2]
- else:
- return max2p5px
+ med = sorted_vals[len(sorted_vals) // 2] if len(sorted_vals) else 1
+ return max2p5px / med
def max2p5pc_med(cell_mask, trap_image):
@@ -68,10 +66,8 @@ def max2p5pc_med(cell_mask, trap_image):
top_vals = sorted_vals[-top_pixels:]
max2p5pc = np.mean(top_vals, dtype=float)
- if max2p5pc:
- return max2p5pc / sorted_vals[len(sorted_vals) // 2]
- else:
- return max2p5pc
+ med = sorted_vals[len(sorted_vals) // 2] if len(sorted_vals) else 1
+ return max2p5pc / med
def std(cell_mask, trap_image):
--
GitLab