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)