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

more debug

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

3
imagelib/color_transfer.py
View file

@ -136,7 +136,6 @@ def linear_color_transfer(target_img, source_img, mode='sym', eps=1e-3):
def lab_image_stats(image, mask=None):
print('np.shape(image):', np.shape(image), 'np.shape(mask):', np.shape(mask))
# compute the mean and standard deviation of each channel
l, a, b = cv2.split(image)
@ -144,6 +143,8 @@ def lab_image_stats(image, mask=None):
im_mask = np.squeeze(mask) if len(np.shape(mask)) == 3 else mask
l, a, b = l[im_mask == 1], a[im_mask == 1], b[im_mask == 1]
print('np.shape(image):', np.shape(image), 'np.shape(im_mask):', np.shape(im_mask))
l_mean, l_std = np.mean(l), np.std(l)
a_mean, a_std = np.mean(a), np.std(a)
b_mean, b_std = np.mean(b), np.std(b)