Skip to content
Snippets Groups Projects
Commit 21e9b5c5 authored by tomrink's avatar tomrink
Browse files

snapshot...

parent 69af6d1e
Branches
Tags
No related merge requests found
...@@ -310,10 +310,12 @@ class ESPCN: ...@@ -310,10 +310,12 @@ class ESPCN:
dataset = dataset.map(self.data_function_evaluate, num_parallel_calls=8) dataset = dataset.map(self.data_function_evaluate, num_parallel_calls=8)
self.eval_dataset = dataset self.eval_dataset = dataset
def setup_pipeline(self, train_data_files, test_data_files, num_train_samples): def setup_pipeline(self, train_data_files, train_label_files, test_data_files, test_label_files, num_train_samples):
self.train_data_files = train_data_files self.train_data_files = train_data_files
self.test_data_files = test_data_files self.test_data_files = test_data_files
self.train_label_files = train_label_files
self.test_label_files = test_label_files
trn_idxs = np.arange(len(train_data_files)) trn_idxs = np.arange(len(train_data_files))
np.random.shuffle(trn_idxs) np.random.shuffle(trn_idxs)
...@@ -633,11 +635,13 @@ class ESPCN: ...@@ -633,11 +635,13 @@ class ESPCN:
def run(self, directory): def run(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')
valid_label_files = glob.glob(directory+'label_valid*.npy')
train_data_files.sort() train_data_files.sort()
valid_data_files.sort() valid_data_files.sort()
self.setup_pipeline(train_data_files, valid_data_files, 100000) self.setup_pipeline(train_data_files, train_label_files, valid_data_files, valid_label_files, 100000)
self.build_model() self.build_model()
self.build_training() self.build_training()
self.build_evaluation() self.build_evaluation()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment