From 484b0f0258b981f1edda28a21a82807b2225ecdd Mon Sep 17 00:00:00 2001 From: Jeremy Hummel Date: Mon, 12 Aug 2019 23:18:12 -0700 Subject: [PATCH] Fix mask check --- samplelib/SampleProcessor.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) 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,