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

clipped A* B* wrong

Browse source
This commit is contained in:
jh 2019-09-12 19:24:54 -07:00
commit 80307cba62
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

@ -154,8 +154,8 @@ def random_color_transform(image, seed=None):
image = image.dot(M) image = image.dot(M)
l, a, b = cv2.split(image) l, a, b = cv2.split(image)
l = np.clip(l, 0, 100) l = np.clip(l, 0, 100)
a = np.clip(a, 0, 100) a = np.clip(a, -127, 127)
b = np.clip(b, 0, 100) b = np.clip(b, -127, 127)
image = cv2.merge([l, a, b]) image = cv2.merge([l, a, b])
image = cv2.cvtColor(image.astype(np.float32), cv2.COLOR_LAB2BGR) image = cv2.cvtColor(image.astype(np.float32), cv2.COLOR_LAB2BGR)
np.clip(image, 0, 1, out=image) np.clip(image, 0, 1, out=image)