From 2b983ad9af85b712d24554cc9b285e38b499bf6d Mon Sep 17 00:00:00 2001 From: Yu-An Chen Date: Wed, 17 May 2023 21:51:46 -0700 Subject: [PATCH] Fix `cv2_downscale_local_mean` call --- palom/img_util.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/palom/img_util.py b/palom/img_util.py index 6beac37..40246e6 100644 --- a/palom/img_util.py +++ b/palom/img_util.py @@ -55,10 +55,11 @@ def entropy_mask(img, kernel_size=9): def is_brightfield_img(img, max_size=100): img = np.array(img) downscale_factor = int(max(img.shape) / max_size) - thumbnail = cv2_downscale_local_mean(img, (downscale_factor, downscale_factor)) - mask = entropy_mask(thumbnail) + if downscale_factor > 1: + img = cv2_downscale_local_mean(img, downscale_factor) + mask = entropy_mask(img) # is using mean better? - return np.median(thumbnail[mask]) < np.median(thumbnail[~mask]) + return np.median(img[mask]) < np.median(img[~mask]) def block_labeled_mask(img_shape, block_shape, out_chunks=None):