Skip to content
Snippets Groups Projects
Commit 7091864b authored by tomrink's avatar tomrink
Browse files

minor...

parent 80c8aba1
Branches
No related tags found
No related merge requests found
...@@ -887,14 +887,7 @@ class UNET: ...@@ -887,14 +887,7 @@ class UNET:
preds = np.argmax(preds, axis=1) preds = np.argmax(preds, axis=1)
self.test_preds = preds self.test_preds = preds
def run(self, filename_l1b_trn, filename_l1b_tst, filename_l2_trn, filename_l2_tst): def run(self, directory):
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):
train_data_files = glob.glob(directory+'data_train*.npy') train_data_files = glob.glob(directory+'data_train*.npy')
valid_data_files = glob.glob(directory+'data_valid*.npy') valid_data_files = glob.glob(directory+'data_valid*.npy')
train_label_files = glob.glob(directory+'label_train*.npy') train_label_files = glob.glob(directory+'label_train*.npy')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment