From 80c8aba1f58a7295b9ccaa8c8ab6eed0a755a7b7 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 25 May 2022 12:14:14 -0500
Subject: [PATCH] minor...
---
modules/deeplearning/unet_l1b_l2.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/modules/deeplearning/unet_l1b_l2.py b/modules/deeplearning/unet_l1b_l2.py
index 764df959..d3084cf3 100644
--- a/modules/deeplearning/unet_l1b_l2.py
+++ b/modules/deeplearning/unet_l1b_l2.py
@@ -214,7 +214,6 @@ class UNET:
# print(e)
def get_in_mem_data_batch(self, idxs, is_training):
- np.sort(idxs)
dat_files = []
lbl_files = []
@@ -236,11 +235,11 @@ class UNET:
label = np.expand_dims(label, axis=3)
else:
for k in idxs:
- f = self.train_data_files[k]
+ f = self.test_data_files[k]
nda = np.load(f)
dat_files.append(nda)
- f = self.train_label_files[k]
+ f = self.test_label_files[k]
nda = np.load(f)
lbl_files.append(nda)
--
GitLab