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

Change scaling

Browse source
This commit is contained in:
Jeremy Hummel 2019-08-12 22:54:20 -07:00
commit c24d6a87ea
1 changed files with 9 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

17
imagelib/color_transfer.py
View file

@ -156,14 +156,15 @@ def _scale_array(arr, clip=True):
if clip:
return np.clip(arr, 0, 1)
mn = arr.min()
mx = arr.max()
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]
return arr
return (arr - np.min(arr)) / np.ptp(arr)
# mn = arr.min()
# mx = arr.max()
# 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]
#
# return arr
def channel_hist_match(source, template, hist_match_threshold=255, mask=None):