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

Updates masking of channels in lab_image_stats

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

6
imagelib/color_transfer.py
View file

@ -138,11 +138,11 @@ def linear_color_transfer(target_img, source_img, mode='pca', eps=1e-5):
def lab_image_stats(image, mask=None):
# compute the mean and standard deviation of each channel
l, a, b = cv2.split(image)
if mask is not None:
mask = np.squeeze(mask)
image = image[mask == 1]
(l, a, b) = cv2.split(image)
l, a, b = l[mask == 1], a[mask == 1], b[mask == 1]
l_mean, l_std = np.mean(l), np.std(l)
a_mean, a_std = np.mean(l), np.std(l)