From b28707b4e8138d106e3f3e7848c866abcd1a4b8a Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Sat, 21 May 2022 11:23:40 -0500
Subject: [PATCH] fix bug in scale method
---
modules/deeplearning/unet_l1b_l2.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/modules/deeplearning/unet_l1b_l2.py b/modules/deeplearning/unet_l1b_l2.py
index 22c3de10..192f60fe 100644
--- a/modules/deeplearning/unet_l1b_l2.py
+++ b/modules/deeplearning/unet_l1b_l2.py
@@ -223,7 +223,7 @@ class UNET:
f = self.train_label_files
nda = np.load(f)
- label = nda[idxs, 0, :, :]
+ label = nda[idxs, label_idx, :, :]
label = np.expand_dims(label, axis=3)
else:
f = self.test_data_files
--
GitLab