diff --git a/samplelib/SampleProcessor.py b/samplelib/SampleProcessor.py index 3e4deee..906d550 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=True) + img_bgr = imagelib.reinhard_color_transfer(img_bgr, ct_sample_bgr, clip=False) 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=True, + clip=False, target_mask=img_mask, source_mask=ct_sample_mask)