github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-22 06:23:20 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix mask check

Browse source
This commit is contained in:
Jeremy Hummel 2019-08-12 23:18:12 -07:00
commit 484b0f0258
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
samplelib/SampleProcessor.py
View file

@ -237,9 +237,10 @@ class SampleProcessor(object):
elif apply_ct == ColorTransferMode.RCT: 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)
elif apply_ct == ColorTransferMode.RCT_MASKED and ct_sample_mask is None: elif apply_ct == ColorTransferMode.RCT_MASKED
ct_sample_mask = ct_sample.load_fanseg_mask() or \ if ct_sample_mask is None:
LandmarksProcessor.get_image_hull_mask(ct_sample_bgr.shape, ct_sample.landmarks) 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, img_bgr = imagelib.reinhard_color_transfer(img_bgr,
ct_sample_bgr, ct_sample_bgr,
clip=True, clip=True,