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

Convert to float32 instead of uint8

Browse source
This commit is contained in:
jh 2021-05-05 11:10:23 -07:00
commit de00810871
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
core/imagelib/color_transfer.py
View file

@ -373,8 +373,8 @@ 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 face = img.astype(np.float32)
face = np.clip(face*255.0, 0, 255).astype(np.uint8) face = np.clip(face*255.0, 0, 255)
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