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

removed fs aug artifacts

Browse source
This commit is contained in:
sinofis 2020-10-15 16:15:14 +02:00
commit 296488be6c
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

@ -374,11 +374,11 @@ def color_transfer(ct_mode, img_src, img_trg):
def color_augmentation(img):
""" Color adjust RGB image """
face = img
face = (face * 255.0).astype("uint8")
face = np.clip(face*255.0, 0, 255).astype(np.uint8)
face = random_clahe(face)
face = random_lab(face)
img[:, :, :3] = face
return np.clip(img.astype('float32') / 255.0, 0, 1)
return (face / 255.0).astype(np.float32)
def random_lab(image):