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

debug

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

2
imagelib/color_transfer.py
View file

@ -136,6 +136,7 @@ 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), 'np.shape(mask):', np.shape(mask))
# 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)
@ -147,6 +148,7 @@ def lab_image_stats(image, mask=None):
a_mean, a_std = np.mean(a), np.std(a) a_mean, a_std = np.mean(a), np.std(a)
b_mean, b_std = np.mean(b), np.std(b) b_mean, b_std = np.mean(b), np.std(b)
print('l_mean, l_std, a_mean, a_std, b_mean, b_std:', l_mean, l_std, a_mean, a_std, b_mean, b_std)
# return the color statistics # return the color statistics
return l_mean, l_std, a_mean, a_std, b_mean, b_std return l_mean, l_std, a_mean, a_std, b_mean, b_std