From ef144182e124e8a48c4ac0eab7b423b71709eed6 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Thu, 15 Jun 2023 11:28:55 -0500 Subject: [PATCH] snapshot... --- modules/deeplearning/cloud_opd_srcnn_abi.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/modules/deeplearning/cloud_opd_srcnn_abi.py b/modules/deeplearning/cloud_opd_srcnn_abi.py index 962bc9db..c2723ff7 100644 --- a/modules/deeplearning/cloud_opd_srcnn_abi.py +++ b/modules/deeplearning/cloud_opd_srcnn_abi.py @@ -208,7 +208,7 @@ class SRCNN: self.test_label_files = None # self.n_chans = len(data_params_half) + len(data_params_full) + 1 - self.n_chans = 3 + self.n_chans = 6 self.X_img = tf.keras.Input(shape=(None, None, self.n_chans)) @@ -267,6 +267,18 @@ class SRCNN: tmp = normalize(tmp, param, mean_std_dct) data_norm.append(tmp) + for param in sub_fields: + idx = params.index(param) + tmp = input_data[:, idx, :, :] + tmp = np.where(np.isnan(tmp), 0, tmp) + tmp = tmp[:, self.slc_y_m, self.slc_x_m] + tmp = self.upsample(tmp) + if param != 'refl_substddev_ch01': + tmp = normalize(tmp, 'refl_0_65um_nom', mean_std_dct) + else: + tmp = np.where(np.isnan(tmp), 0, tmp) + data_norm.append(tmp) + # for param in sub_fields: # idx = params.index(param) # tmp = input_data[:, idx, :, :] -- GitLab