From ba5e1c265e0a1d09fd2d320d910facfa91692cff Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Sun, 2 Oct 2022 09:44:45 -0500
Subject: [PATCH] snapshot...

---
 modules/deeplearning/srcnn_l1b_l2.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py
index 560e9f61..db1f7aad 100644
--- a/modules/deeplearning/srcnn_l1b_l2.py
+++ b/modules/deeplearning/srcnn_l1b_l2.py
@@ -221,7 +221,7 @@ class SRCNN:
             tmp = normalize(tmp, param, mean_std_dct, add_noise=add_noise, noise_scale=noise_scale)
             data_norm.append(tmp)
         # --------
-        tmp = input_data[:, 2, 3:131:2, 3:131:2]
+        tmp = input_data[:, label_idx, 3:131:2, 3:131:2]
         tmp = resample_2d_linear(x_64, y_64, tmp, t, s)
         #tmp = normalize(tmp, 'temp_11_0um_nom', mean_std_dct)
         data_norm.append(tmp)
-- 
GitLab