diff --git a/samplelib/SampleProcessor.py b/samplelib/SampleProcessor.py index 906d550..cbbbe47 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=False, preserve_paper=True,) elif apply_ct == ColorTransferMode.RCT_MASKED: if ct_sample_mask is None: @@ -244,6 +244,7 @@ class SampleProcessor(object): img_bgr = imagelib.reinhard_color_transfer(img_bgr, ct_sample_bgr, clip=False, + preserve_paper=True, target_mask=img_mask, source_mask=ct_sample_mask)