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

needs float32

Browse source
This commit is contained in:
jh 2019-09-12 19:08:02 -07:00
commit 1ef2716293
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
imagelib/color_transfer.py
View file

@ -148,7 +148,7 @@ def seamless_clone(source, target, mask):
def random_color_transform(image, seed=None):
image = cv2.cvtColor(image, cv2.COLOR_BGR2LAB)
image = cv2.cvtColor(image.astype(np.float32), cv2.COLOR_BGR2LAB)
M = special_ortho_group.rvs(3, 1, seed)
image = image.dot(M)
l, a, b = cv2.split(image)