From 7091864b3714529ca22ee5728e931cb84b56d9b3 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Wed, 25 May 2022 12:17:17 -0500 Subject: [PATCH] minor... --- modules/deeplearning/unet_l1b_l2.py | 9 +-------- 1 file changed, 1 insertion(+), 8 deletions(-) diff --git a/modules/deeplearning/unet_l1b_l2.py b/modules/deeplearning/unet_l1b_l2.py index d3084cf3..e4a025a6 100644 --- a/modules/deeplearning/unet_l1b_l2.py +++ b/modules/deeplearning/unet_l1b_l2.py @@ -887,14 +887,7 @@ class UNET: preds = np.argmax(preds, axis=1) self.test_preds = preds - def run(self, filename_l1b_trn, filename_l1b_tst, filename_l2_trn, filename_l2_tst): - self.setup_pipeline(filename_l1b_trn, filename_l1b_tst, filename_l2_trn, filename_l2_tst) - self.build_model() - self.build_training() - self.build_evaluation() - self.do_training() - - def run_test(self, directory): + def run(self, directory): train_data_files = glob.glob(directory+'data_train*.npy') valid_data_files = glob.glob(directory+'data_valid*.npy') train_label_files = glob.glob(directory+'label_train*.npy') -- GitLab