From f697cc24a0ad8087d0660d31fe6c30e121033cec Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 21 Jun 2023 13:22:22 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/cloud_opd_srcnn_abi.py | 11 ++++++++---
1 file changed, 8 insertions(+), 3 deletions(-)
diff --git a/modules/deeplearning/cloud_opd_srcnn_abi.py b/modules/deeplearning/cloud_opd_srcnn_abi.py
index 730207e3..24384512 100644
--- a/modules/deeplearning/cloud_opd_srcnn_abi.py
+++ b/modules/deeplearning/cloud_opd_srcnn_abi.py
@@ -658,8 +658,12 @@ class SRCNN:
def run(self, directory, ckpt_dir=None, num_data_samples=50000):
train_data_files = glob.glob(directory+'train*mres*.npy')
valid_data_files = glob.glob(directory+'valid*mres*.npy')
- train_label_files = glob.glob(directory+'train*ires*.npy')
- valid_label_files = glob.glob(directory+'valid*ires*.npy')
+
+ # train_label_files = glob.glob(directory+'train*ires*.npy')
+ # valid_label_files = glob.glob(directory+'valid*ires*.npy')
+
+ train_label_files = [f.replace('mres', 'ires'), f in train_data_files]
+ valid_label_files = [f.replace('mres', 'ires'), f in valid_data_files]
self.setup_pipeline(train_data_files, train_label_files, valid_data_files, valid_label_files, num_data_samples)
self.build_model()
@@ -671,7 +675,8 @@ class SRCNN:
self.num_data_samples = 1000
valid_data_files = glob.glob(directory + 'valid*mres*.npy')
- valid_label_files = glob.glob(directory + 'valid*ires*.npy')
+ # valid_label_files = glob.glob(directory + 'valid*ires*.npy')
+ valid_label_files = [f.replace('mres', 'ires'), f in valid_data_files]
self.setup_test_pipeline(valid_data_files, valid_label_files)
self.build_model()
--
GitLab