github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-22 22:34:25 -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): def _scale_array(arr, clip=True):
if clip: if clip:
return np.clip(arr, 0, 255) return np.clip(arr, 0, 1)
mn = arr.min() mn = arr.min()
mx = arr.max() 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]: if mn < scale_range[0] or mx > scale_range[1]:
return (scale_range[1] - scale_range[0]) * (arr - mn) / (mx - mn) + scale_range[0] return (scale_range[1] - scale_range[0]) * (arr - mn) / (mx - mn) + scale_range[0]