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

snapshot...

parent e436d413
No related branches found
No related tags found
No related merge requests found
......@@ -343,16 +343,6 @@ class SRCNN:
data = data.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
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