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

convert float32

Browse source
This commit is contained in:
jh 2021-05-05 11:41:48 -07:00
commit 00841ab8c0
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
core/imagelib/color_transfer.py
View file

@ -373,8 +373,9 @@ 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 """
face = img.astype(np.float32) img = img.astype(np.float32)
face = np.clip(face*255.0, 0, 255) face = img
face = np.clip(face*255.0, 0, 255).astype(np.uint8)
face = random_clahe(face, seed) face = random_clahe(face, seed)
face = random_lab(face, seed) face = random_lab(face, seed)
img[:, :, :3] = face img[:, :, :3] = face