diff --git a/extraction/core/functions/cell.py b/extraction/core/functions/cell.py index 77179cc4f67183b9180d7ffd12e0a50e6ba7a4b6..cd52aeb3ebc216315b17adbee480f0f5f2315502 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):