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

back to float

Browse source
This commit is contained in:
Jeremy Hummel 2019-08-13 00:53:47 -07:00
commit 3ae562189d
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
imagelib/color_transfer.py
View file

@ -84,7 +84,7 @@ def reinhard_color_transfer(target, source, clip=False, preserve_paper=False, ta
# merge the channels together and convert back to the RGB color
transfer = cv2.merge([l, a, b])
transfer = cv2.cvtColor(transfer.astype("uint8"), cv2.COLOR_LAB2BGR)
transfer = cv2.cvtColor(transfer.astype("uint8"), cv2.COLOR_LAB2BGR).astype("float32")
np.clip(transfer, 0, 1, out=transfer)
# return the color transferred image