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):
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)
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)
l -= min(l_min, 0)
a -= min(a_min, -127)
b -= min(b_min, -127)
a -= min(a_min, 0)
b -= min(b_min, 0)
if l_ptp > 100:
l = 100 * l / l_ptp
if a_ptp > 254:
a = 254 * a / a_ptp - 127
if l_ptp > 255:
l = 255 * l / l_ptp
if a_ptp > 255:
a = 255 * a / a_ptp
if b_ptp:
b = 254 * b / b_ptp - 127
b = 255 * b / b_ptp
return l, a, b