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