clip = false

This commit is contained in:
Jeremy Hummel 2019-08-13 01:23:03 -07:00
commit 6c6065c020

View file

@ -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)