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

Fix scale to use float32

Browse source 
Using float32 instead of unit8, scale 0..1 instead of 0..255
This commit is contained in:
Jeremy Hummel 2019-08-12 22:41:53 -07:00
commit 01c64dadf3
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
imagelib/color_transfer.py
View file

@ -154,11 +154,11 @@ def lab_image_stats(image, mask=None):
def _scale_array(arr, clip=True):
if clip:
return np.clip(arr, 0, 255)
return np.clip(arr, 0, 1)
mn = arr.min()
mx = arr.max()
scale_range = (max([mn, 0]), min([mx, 255]))
scale_range = (max([mn, 0]), min([mx, 1]))
if mn < scale_range[0] or mx > scale_range[1]:
return (scale_range[1] - scale_range[0]) * (arr - mn) / (mx - mn) + scale_range[0]