From 0b82691338892ac2752e6a9d8338dc2609a1abad Mon Sep 17 00:00:00 2001 From: Jeremy Hummel Date: Wed, 14 Aug 2019 12:52:57 -0700 Subject: [PATCH] more debug --- imagelib/color_transfer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/imagelib/color_transfer.py b/imagelib/color_transfer.py index a143381..33dae19 100644 --- a/imagelib/color_transfer.py +++ b/imagelib/color_transfer.py @@ -136,7 +136,6 @@ def linear_color_transfer(target_img, source_img, mode='sym', eps=1e-3): def lab_image_stats(image, mask=None): - print('np.shape(image):', np.shape(image), 'np.shape(mask):', np.shape(mask)) # compute the mean and standard deviation of each channel l, a, b = cv2.split(image) @@ -144,6 +143,8 @@ def lab_image_stats(image, mask=None): im_mask = np.squeeze(mask) if len(np.shape(mask)) == 3 else mask l, a, b = l[im_mask == 1], a[im_mask == 1], b[im_mask == 1] + print('np.shape(image):', np.shape(image), 'np.shape(im_mask):', np.shape(im_mask)) + l_mean, l_std = np.mean(l), np.std(l) a_mean, a_std = np.mean(a), np.std(a) b_mean, b_std = np.mean(b), np.std(b)