Skip to content
Snippets Groups Projects
Commit 3f71dff0 authored by tomrink's avatar tomrink
Browse files

snapshot...

parent e436d413
Branches
No related tags found
No related merge requests found
...@@ -343,16 +343,6 @@ class SRCNN: ...@@ -343,16 +343,6 @@ class SRCNN:
data = data.astype(np.float32) data = data.astype(np.float32)
label = label.astype(np.float32) label = label.astype(np.float32)
if is_training and DO_AUGMENT:
data_ud = np.flip(data, axis=1)
label_ud = np.flip(label, axis=1)
data_lr = np.flip(data, axis=2)
label_lr = np.flip(label, axis=2)
data = np.concatenate([data, data_ud, data_lr])
label = np.concatenate([label, label_ud, label_lr])
return data, label return data, label
def get_in_mem_data_batch_train(self, idxs): def get_in_mem_data_batch_train(self, idxs):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment