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 bad log line

Browse source
This commit is contained in:
Jeremy Hummel 2019-08-14 13:03:49 -07:00
commit 5f7f1fc5e0
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
imagelib/color_transfer.py
View file

@ -136,14 +136,14 @@ def linear_color_transfer(target_img, source_img, mode='sym', eps=1e-3):
def lab_image_stats(image, mask=None): def lab_image_stats(image, mask=None):
print('np.shape(image):', np.shape(image), )
# compute the mean and standard deviation of each channel # compute the mean and standard deviation of each channel
l, a, b = cv2.split(image) l, a, b = cv2.split(image)
if mask is not None: if mask is not None:
im_mask = np.squeeze(mask) if len(np.shape(mask)) == 3 else mask 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] l, a, b = l[im_mask == 1], a[im_mask == 1], b[im_mask == 1]
print('np.shape(im_mask):', np.shape(im_mask))
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) l_mean, l_std = np.mean(l), np.std(l)
a_mean, a_std = np.mean(a), np.std(a) a_mean, a_std = np.mean(a), np.std(a)