From 98e2ede1b24dd0169ab5d2218d430e9ded4f7808 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 22 Jun 2023 13:46:09 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/cloud_opd_srcnn_viirs.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/modules/deeplearning/cloud_opd_srcnn_viirs.py b/modules/deeplearning/cloud_opd_srcnn_viirs.py
index 9fca0151..19653240 100644
--- a/modules/deeplearning/cloud_opd_srcnn_viirs.py
+++ b/modules/deeplearning/cloud_opd_srcnn_viirs.py
@@ -653,8 +653,8 @@ 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 = [f.replace('mres', 'ires') for f in train_data_files]
+ valid_label_files = [f.replace('mres', 'ires') for 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()
@@ -666,7 +666,7 @@ 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 = [f.replace('mres', 'ires') for f in valid_data_files]
self.setup_test_pipeline(valid_data_files, valid_label_files)
self.build_model()
--
GitLab