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

Merge pull request #141 from faceshiftlabs/fix/fs-aug

Browse source 
Fix/fs aug
This commit is contained in:
Jeremy Hummel 2021-05-05 11:49:09 -07:00 • committed by GitHub
commit 0b5ceff405
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
core/imagelib/color_transfer.py
View file

@ -373,6 +373,7 @@ def color_transfer(ct_mode, img_src, img_trg):
# imported from faceswap # imported from faceswap
def color_augmentation(img, seed=None): def color_augmentation(img, seed=None):
""" Color adjust RGB image """ """ Color adjust RGB image """
img = img.astype(np.float32)
face = img face = img
face = np.clip(face*255.0, 0, 255).astype(np.uint8) face = np.clip(face*255.0, 0, 255).astype(np.uint8)
face = random_clahe(face, seed) face = random_clahe(face, seed)