From 1d49825b0f8a585654129858e09f14c8ce48ca94 Mon Sep 17 00:00:00 2001 From: Jeremy Hummel Date: Tue, 13 Aug 2019 00:31:28 -0700 Subject: [PATCH] turn on clipping --- samplelib/SampleProcessor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/samplelib/SampleProcessor.py b/samplelib/SampleProcessor.py index 906d550..3e4deee 100644 --- a/samplelib/SampleProcessor.py +++ b/samplelib/SampleProcessor.py @@ -235,7 +235,7 @@ class SampleProcessor(object): img_bgr = imagelib.linear_color_transfer(img_bgr, ct_sample_bgr) elif apply_ct == ColorTransferMode.RCT: - img_bgr = imagelib.reinhard_color_transfer(img_bgr, ct_sample_bgr, clip=False) + img_bgr = imagelib.reinhard_color_transfer(img_bgr, ct_sample_bgr, clip=True) elif apply_ct == ColorTransferMode.RCT_MASKED: if ct_sample_mask is None: @@ -243,7 +243,7 @@ class SampleProcessor(object): LandmarksProcessor.get_image_hull_mask(ct_sample_bgr.shape, ct_sample.landmarks) img_bgr = imagelib.reinhard_color_transfer(img_bgr, ct_sample_bgr, - clip=False, + clip=True, target_mask=img_mask, source_mask=ct_sample_mask)