From 6931206edf7f187f01f1ec31e895a38726e8aec3 Mon Sep 17 00:00:00 2001 From: Jin Igarashi Date: Tue, 9 Apr 2024 18:31:29 +0900 Subject: [PATCH] refactor: use only a valid_mask for simplifying (#91) --- src/cogserver/algorithms/rca.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/cogserver/algorithms/rca.py b/src/cogserver/algorithms/rca.py index d1361c7..cc73a02 100644 --- a/src/cogserver/algorithms/rca.py +++ b/src/cogserver/algorithms/rca.py @@ -79,9 +79,12 @@ def __call__(self, img: ImageData) -> ImageData: v = 0.1 #v = data.ptp()*.1 datam = (data > -v) & (data < v) + valid_mask |= datam if self.threshold: - datam = datam | np.abs(data)< self.threshold - arr = numpy.ma.masked_array(data*100, dtype=self.output_dtype, mask=valid_mask | datam ) + threshold_mask = np.abs(data)< self.threshold + valid_mask |= threshold_mask + + arr = numpy.ma.masked_array(data*100, dtype=self.output_dtype, mask=valid_mask) return ImageData( arr, assets=img.assets,