From 3541ac51e883c7d2d5ff960e3fc7d946defb214a Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 17 Mar 2022 15:10:56 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_fcn.py | 10 ++++++++++
1 file changed, 10 insertions(+)
diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index a6d9f862..ce9b188e 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -34,6 +34,7 @@ CONV3D = False
NOISE_TRAINING = False
NOISE_STDDEV = 0.10
+DO_AUGMENT = False
img_width = 16
@@ -304,6 +305,15 @@ class IcingIntensityFCN:
else:
self.in_mem_data_cache_test[key] = (data, data_alt, label)
+ if is_training and DO_AUGMENT:
+ data_ud = np.flip(data, axis=1)
+ data_alt_ud = np.copy(data_alt)
+ label_ud = np.copy(label)
+
+ data = np.concatenate([data, data_ud])
+ data_alt = np.concatenate([data_alt, data_alt_ud])
+ label = np.concatenate([label, label_ud])
+
return data, data_alt, label
def get_parameter_data(self, param, nd_idxs, is_training):
--
GitLab