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

only rotate over A*B*

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

3
imagelib/color_transfer.py
View file

@ -149,7 +149,8 @@ def seamless_clone(source, target, mask):
def random_color_transform(image, seed=None):
image = cv2.cvtColor(image, cv2.COLOR_BGR2LAB)
M = special_ortho_group.rvs(3, 1, seed)
M = np.eye(3)
M[1:, 1:] = special_ortho_group.rvs(2, 1, seed)
image = image.dot(M)
l, a, b = cv2.split(image)
l = np.clip(l, 0, 100)