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

debugging converter

Browse source
This commit is contained in:
Jeremy Hummel 2019-08-14 12:09:18 -07:00
commit 2c7e4560be
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
samplelib/SampleProcessor.py
View file

@ -259,6 +259,7 @@ class SampleProcessor(object):
else: else:
if ct_sample_mask is None: if ct_sample_mask is None:
ct_sample_mask = ct_sample.load_mask() ct_sample_mask = ct_sample.load_mask()
print(np.shape(img_bgr), np.shape(img_mask), np.shape(ct_sample_bgr), np.shape(ct_sample_mask))
img_bgr = imagelib.reinhard_color_transfer(img_bgr, ct_sample_bgr, clip=use_clip, img_bgr = imagelib.reinhard_color_transfer(img_bgr, ct_sample_bgr, clip=use_clip,
preserve_paper=use_paper, source_mask=img_mask, preserve_paper=use_paper, source_mask=img_mask,
target_mask=ct_sample_mask) target_mask=ct_sample_mask)