From 62991ccc4a507cdacc368a57ccb8034efd10a3e3 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 26 Sep 2022 09:50:23 -0500
Subject: [PATCH] minor
---
modules/deeplearning/srcnn.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/modules/deeplearning/srcnn.py b/modules/deeplearning/srcnn.py
index 5f2146ea..cd3881a7 100644
--- a/modules/deeplearning/srcnn.py
+++ b/modules/deeplearning/srcnn.py
@@ -194,13 +194,13 @@ class SRCNN:
def get_in_mem_data_batch(self, idxs, is_training):
if is_training:
- label_files = self.train_data_files
+ files = self.train_data_files
else:
- label_files = self.test_data_files
+ files = self.test_data_files
label_s = []
for k in idxs:
- f = label_files[k]
+ f = files[k]
nda = np.load(f)
label_s.append(nda)
--
GitLab