diff --git a/samplelib/SampleProcessor.py b/samplelib/SampleProcessor.py index 3e4deee..a5dbb1b 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=True, 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=True, + preserve_paper=True, target_mask=img_mask, source_mask=ct_sample_mask)