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

fix scaling

Browse source
This commit is contained in:
Jeremy Hummel 2019-08-13 00:26:22 -07:00
commit be5585a3b4
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

@ -152,7 +152,8 @@ def lab_image_stats(image, mask=None):
def _scale_array(l, a, b, clip=True): def _scale_array(l, a, b, clip=True):
if clip: if clip:
return np.clip(l, 0, 100), np.clip(a, -127, 127), np.clip(b, -127, 127) return np.clip(l, 0, 255), np.clip(a, 0, 255), np.clip(b, 0, 255)
# return np.clip(l, 0, 100), np.clip(a, -127, 127), np.clip(b, -127, 127)
# return (arr - np.min(arr)) / np.ptp(arr) # return (arr - np.min(arr)) / np.ptp(arr)
l_min, a_min, b_min = np.min(l), np.min(a), np.min(b) l_min, a_min, b_min = np.min(l), np.min(a), np.min(b)
@ -163,15 +164,15 @@ def _scale_array(l, a, b, clip=True):
print('l*a*b* max:', l_max, a_max, b_max) print('l*a*b* max:', l_max, a_max, b_max)
l -= min(l_min, 0) l -= min(l_min, 0)
a -= min(a_min, -127) a -= min(a_min, 0)
b -= min(b_min, -127) b -= min(b_min, 0)
if l_ptp > 100: if l_ptp > 255:
l = 100 * l / l_ptp l = 255 * l / l_ptp
if a_ptp > 254: if a_ptp > 255:
a = 254 * a / a_ptp - 127 a = 255 * a / a_ptp
if b_ptp: if b_ptp:
b = 254 * b / b_ptp - 127 b = 255 * b / b_ptp
return l, a, b return l, a, b