From 12a4b7788fc590cf3e4a42098eb013ff41801c83 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 14 May 2021 11:13:16 -0500
Subject: [PATCH] minor...
---
modules/deeplearning/icing_cnn.py | 19 -------------------
1 file changed, 19 deletions(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 67801601..18d94b31 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -215,25 +215,6 @@ class IcingIntensityNN:
label = label.astype(np.int32)
label = np.where(label == -1, 0, label)
- # # Augmentation, TODO: work into Dataset.map (efficiency)
- # data_aug = []
- # label_aug = []
- # for k in range(label.shape[0]):
- # if label[k] == 3 or label[k] == 4 or label[k] == 5 or label[k] == 6:
- # data_aug.append(tf.image.flip_up_down(data[k,]).numpy())
- # data_aug.append(tf.image.flip_left_right(data[k,]).numpy())
- # data_aug.append(tf.image.rot90(data[k,]).numpy())
- # label_aug.append(label[k])
- # label_aug.append(label[k])
- # label_aug.append(label[k])
- #
- # data_aug = np.stack(data_aug)
- # label_aug = np.stack(label_aug)
- #
- # data = np.concatenate([data, data_aug])
- # label = np.concatenate([label, label_aug])
- # # --------------------------------------------------------
-
# binary, two class
if NumClasses == 2:
label = np.where(label != 0, 1, label)
--
GitLab