diff --git a/modules/util/util.py b/modules/util/util.py index 993b8edc483c662f776876d37f6f6c2696d7159c..7cd3cb7c6a2bcf55416029c59825151f2f13275d 100644 --- a/modules/util/util.py +++ b/modules/util/util.py @@ -589,6 +589,24 @@ def scale(data, param, mean_std_dict, copy=True): return data +def scale2(data, lo, hi, copy=True): + if copy: + data = data.copy() + + shape = data.shape + data = data.flatten() + + data -= lo + data /= (hi - lo) + + not_valid = np.isnan(data) + data[not_valid] = 0 + + data = np.reshape(data, shape) + + return data + + def descale(data, param, mean_std_dict, copy=True): if copy: data = data.copy()