From 999d6ff7be439d9dd782ec11dceeaa6441712d04 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 9 Jul 2021 11:08:07 -0500
Subject: [PATCH] minor
---
modules/deeplearning/icing_cnn.py | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 818f2f49..90477968 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -27,6 +27,8 @@ EARLY_STOP = True
TRIPLET = False
CONV3D = False
+NOISE_TRAINING = False
+
img_width = 16
mean_std_file = homedir+'data/icing/mean_std_l1b_no_ice.pkl'
@@ -211,7 +213,10 @@ class IcingIntensityNN:
data = []
for param in train_params:
nda = h5f[param][nd_idxs, ]
- nda = normalize(nda, param, mean_std_dct)
+ if NOISE_TRAINING and is_training:
+ nda = normalize(nda, param, mean_std_dct, add_noise=True, noise_scale=0.01, seed=42)
+ else:
+ nda = normalize(nda, param, mean_std_dct)
data.append(nda)
data = np.stack(data)
data = data.astype(np.float32)
--
GitLab