From c24d6a87ea3f23a0d25d357a29f738c5dea80c03 Mon Sep 17 00:00:00 2001 From: Jeremy Hummel Date: Mon, 12 Aug 2019 22:54:20 -0700 Subject: [PATCH] Change scaling --- imagelib/color_transfer.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/imagelib/color_transfer.py b/imagelib/color_transfer.py index b05035b..dd190cf 100644 --- a/imagelib/color_transfer.py +++ b/imagelib/color_transfer.py @@ -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):