From a4371edcd681eac7a25edbdf1d40c1c359a2fcc3 Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Thu, 12 Sep 2024 16:15:55 +0200
Subject: [PATCH] Merged in smoothing operation that supports binary masks
---
model_server/base/process.py | 11 +++++++++--
1 file changed, 9 insertions(+), 2 deletions(-)
diff --git a/model_server/base/process.py b/model_server/base/process.py
index d475b063..dac347cd 100644
--- a/model_server/base/process.py
+++ b/model_server/base/process.py
@@ -18,7 +18,7 @@ def is_mask(img):
return True
elif img.dtype == 'uint8':
unique = np.unique(img)
- if unique.shape[0] == 2 and np.all(unique == [0, 255]):
+ if unique.shape[0] <= 2 and np.all(unique == [0, 255]):
return True
return False
@@ -136,7 +136,14 @@ def smooth(img: np.ndarray, sig: float) -> np.ndarray:
:param sig: threshold parameter
:return: smoothed image
"""
- return gaussian(img, sig)
+ ga = gaussian(img, sig, preserve_range=True)
+ if is_mask(img):
+ if img.dtype == 'bool':
+ return ga > ga.mean()
+ elif img.dtype == 'uint8':
+ return (255 * (ga > ga.mean())).astype('uint8')
+ else:
+ return ga
class Error(Exception):
pass
--
GitLab