From ca63ee2d993c86aef951d3e65613d4ccf562a961 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 20 May 2022 14:40:12 -0500
Subject: [PATCH] add method to scale data
---
modules/deeplearning/unet_l1b_l2.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/unet_l1b_l2.py b/modules/deeplearning/unet_l1b_l2.py
index 06d7a55d..c9145883 100644
--- a/modules/deeplearning/unet_l1b_l2.py
+++ b/modules/deeplearning/unet_l1b_l2.py
@@ -60,7 +60,8 @@ l2_params = ['cloud_fraction', 'cld_temp_acha', 'cld_press_acha']
zero_out_params = ['cld_reff_dcomp', 'cld_opd_dcomp', 'iwc_dcomp', 'lwc_dcomp']
DO_ZERO_OUT = False
-label_param = l2_params[1]
+label_idx = 1
+label_param = l2_params[label_idx]
def build_conv2d_block(conv, num_filters, activation, block_name, padding='SAME'):
@@ -231,7 +232,7 @@ class UNET:
f = self.test_label_files
nda = np.load(f)
- label = nda[idxs, 0, :, :]
+ label = nda[idxs, label_idx, :, :]
label = np.expand_dims(label, axis=3)
data = data.astype(np.float32)
--
GitLab