From 2caa70ca00c67cc08f9d7e858c9badf37641ca48 Mon Sep 17 00:00:00 2001 From: Jeremy Hummel Date: Sun, 11 Aug 2019 16:28:11 -0700 Subject: [PATCH] Adds masking to Sample Processor --- imagelib/color_transfer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/imagelib/color_transfer.py b/imagelib/color_transfer.py index eb85356..67a580d 100644 --- a/imagelib/color_transfer.py +++ b/imagelib/color_transfer.py @@ -51,8 +51,8 @@ def reinhard_color_transfer(target, source, clip=False, preserve_paper=False, ta target = cv2.cvtColor(target.astype(np.float32), cv2.COLOR_BGR2LAB) # compute color statistics for the source and target images - (lMeanSrc, lStdSrc, aMeanSrc, aStdSrc, bMeanSrc, bStdSrc) = lab_image_stats(src_input) - (lMeanTar, lStdTar, aMeanTar, aStdTar, bMeanTar, bStdTar) = lab_image_stats(tgt_input) + (lMeanSrc, lStdSrc, aMeanSrc, aStdSrc, bMeanSrc, bStdSrc) = lab_image_stats(source, mask=source_mask) + (lMeanTar, lStdTar, aMeanTar, aStdTar, bMeanTar, bStdTar) = lab_image_stats(target, mask=target_mask) # subtract the means from the target image (l, a, b) = cv2.split(target)