From 401bfc0283bf22b5b36a01aced0b07a756ad66f0 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 27 Sep 2022 11:01:44 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/srcnn_l1b_l2.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py
index 60fdb546..9932cfbf 100644
--- a/modules/deeplearning/srcnn_l1b_l2.py
+++ b/modules/deeplearning/srcnn_l1b_l2.py
@@ -212,8 +212,9 @@ class SRCNN:
noise_scale = 0.005
data_norm = []
- for k, param in enumerate(params):
- tmp = data[:, k, 3:131:2, 3:131:2]
+ for param in data_params:
+ idx = params.index(param)
+ tmp = data[:, idx, 3:131:2, 3:131:2]
tmp = resample(y_64, x_64, tmp, s, t)
tmp = normalize(tmp, param, mean_std_dct, add_noise=add_noise, noise_scale=noise_scale)
data_norm.append(tmp)
--
GitLab