From 23b4149ae25bc0b8e090ff2e1a217f80fe1ff263 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 27 Oct 2022 14:28:01 -0500
Subject: [PATCH] minor...
---
modules/deeplearning/srcnn_l1b_l2.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py
index 899f68dc..8a43aea3 100644
--- a/modules/deeplearning/srcnn_l1b_l2.py
+++ b/modules/deeplearning/srcnn_l1b_l2.py
@@ -230,7 +230,8 @@ class SRCNN:
tmp = resample_2d_linear(x_64, y_64, tmp, t, s)
data_norm.append(tmp)
# --------
- idx = params.index('refl_0_65um_nom')
+ param = 'refl_0_65um_nom'
+ idx = params.index(param)
tmp = input_data[:, idx, 3:131:2, 3:131:2]
# tmp = input_data[:, idx, 3:131, 3:131]
tmp = normalize(tmp, param, mean_std_dct, add_noise=add_noise, noise_scale=noise_scale)
@@ -239,7 +240,7 @@ class SRCNN:
# --------
tmp = input_data[:, label_idx, 3:131:2, 3:131:2]
if label_param != 'cloud_fraction':
- tmp = normalize(tmp, param, mean_std_dct, add_noise=add_noise, noise_scale=noise_scale)
+ tmp = normalize(tmp, label_param, mean_std_dct, add_noise=add_noise, noise_scale=noise_scale)
else:
tmp = np.where(np.isnan(tmp), 0, tmp)
tmp = resample_2d_linear(x_64, y_64, tmp, t, s)
@@ -247,7 +248,7 @@ class SRCNN:
# ---------
data = np.stack(data_norm, axis=3)
data = data.astype(np.float32)
-
+ # -----------------------------------------------------
# label = input_data[:, label_idx, 3:131:2, 3:131:2]
label = input_data[:, label_idx, 3:131, 3:131]
if label_param != 'cloud_fraction':
--
GitLab