From a8566ff699a1aec2d278b534d9e69fd6fd1a6ccb Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 7 Dec 2022 11:07:25 -0600
Subject: [PATCH] snapshot..
---
modules/util/util.py | 30 +++++++++++++++++++++---------
1 file changed, 21 insertions(+), 9 deletions(-)
diff --git a/modules/util/util.py b/modules/util/util.py
index 0143f4a5..44a034e2 100644
--- a/modules/util/util.py
+++ b/modules/util/util.py
@@ -435,13 +435,14 @@ def make_histogram(values, edges):
return h
-def normalize(data, param, mean_std_dict, add_noise=False, noise_scale=0.01, seed=None, copy=True):
- if copy:
- data = data.copy()
+def normalize(data, param, mean_std_dict, copy=True):
if mean_std_dict.get(param) is None:
return data
+ if copy:
+ data = data.copy()
+
shape = data.shape
data = data.flatten()
@@ -449,12 +450,6 @@ def normalize(data, param, mean_std_dict, add_noise=False, noise_scale=0.01, see
data -= mean
data /= std
- if add_noise:
- if seed is not None:
- np.random.seed(seed)
- rnd = np.random.normal(loc=0, scale=noise_scale, size=data.size)
- data += rnd
-
not_valid = np.isnan(data)
data[not_valid] = 0
@@ -463,6 +458,23 @@ def normalize(data, param, mean_std_dict, add_noise=False, noise_scale=0.01, see
return data
+def add_noise(data, noise_scale=0.01, seed=None, copy=True):
+ if copy:
+ data = data.copy()
+
+ shape = data.shape
+ data = data.flatten()
+
+ if seed is not None:
+ np.random.seed(seed)
+ rnd = np.random.normal(loc=0, scale=noise_scale, size=data.size)
+ data += rnd
+
+ data = np.reshape(data, shape)
+
+ return data
+
+
def denormalize(data, param, mean_std_dict, copy=True):
if copy:
data = data.copy()
--
GitLab