From b58ed2d0eef91ecb85951750d3a8ee2b8e9ae438 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Sun, 4 Feb 2024 11:15:59 -0600 Subject: [PATCH] snapshot... --- modules/deeplearning/icing_cnn.py | 24 ++++++++++++------------ 1 file changed, 12 insertions(+), 12 deletions(-) diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py index c07277d1..2da79d32 100644 --- a/modules/deeplearning/icing_cnn.py +++ b/modules/deeplearning/icing_cnn.py @@ -318,18 +318,18 @@ class IcingIntensityNN: 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_lr = np.flip(data, axis=2) - data_alt_lr = np.copy(data_alt) - label_lr = np.copy(label) - - data = np.concatenate([data, data_ud, data_lr]) - data_alt = np.concatenate([data_alt, data_alt_ud, data_alt_lr]) - label = np.concatenate([label, label_ud, label_lr]) + # 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_lr = np.flip(data, axis=2) + # data_alt_lr = np.copy(data_alt) + # label_lr = np.copy(label) + # + # data = np.concatenate([data, data_ud, data_lr]) + # data_alt = np.concatenate([data_alt, data_alt_ud, data_alt_lr]) + # label = np.concatenate([label, label_ud, label_lr]) return data, data_alt, label -- GitLab