diff --git a/samplelib/SampleProcessor.py b/samplelib/SampleProcessor.py index dab8bc3..635066e 100644 --- a/samplelib/SampleProcessor.py +++ b/samplelib/SampleProcessor.py @@ -237,9 +237,10 @@ class SampleProcessor(object): elif apply_ct == ColorTransferMode.RCT: img_bgr = imagelib.reinhard_color_transfer(img_bgr, ct_sample_bgr, clip=True) - elif apply_ct == ColorTransferMode.RCT_MASKED and ct_sample_mask is None: - ct_sample_mask = ct_sample.load_fanseg_mask() or \ - LandmarksProcessor.get_image_hull_mask(ct_sample_bgr.shape, ct_sample.landmarks) + elif apply_ct == ColorTransferMode.RCT_MASKED + if ct_sample_mask is None: + ct_sample_mask = ct_sample.load_fanseg_mask() or \ + 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,