Skip to content
Snippets Groups Projects
Commit 5f7eed82 authored by tomrink's avatar tomrink
Browse files

snapshot...

parent 6ba6865e
No related merge requests found
......@@ -658,10 +658,6 @@ 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)
......@@ -675,7 +671,6 @@ 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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment