snapshot...

modules/deeplearning/cnn_cld_frac_mod_res.py

@@ -65,8 +65,8 @@ params = ['temp_11_0um_nom', 'refl_0_65um_nom', label_param]
 data_params_half = ['temp_11_0um_nom']
 data_params_full = ['refl_0_65um_nom']
 
-# label_idx = params.index(label_param)
-label_idx = 0
+label_idx = params.index(label_param)
+# label_idx = 0
 
 print('data_params_half: ', data_params_half)
 print('data_params_full: ', data_params_full)
@@ -350,7 +350,7 @@ class SRCNN:
             label_s.append(nda)
         input_data = np.concatenate(data_s)
         input_label = np.concatenate(label_s)
-        input_label = input_label[:, label_idx, :, :]
+        input_label = input_label[:, 0, :, :]
 
         data_norm = []
         for param in data_params_half:
@@ -381,7 +381,7 @@ class SRCNN:
             data_norm.append(avg[:, 0:66, 0:66])
             # data_norm.append(std[:, 0:66, 0:66])
         # ---------------------------------------------------
-        tmp = input_label
+        tmp = input_data[:, label_idx, :, :]
         tmp = tmp.copy()
         tmp = np.where(np.isnan(tmp), 0, tmp)
         if DO_ESPCN: