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

snapshot...

parent f8996692
Branches
No related tags found
No related merge requests found
......@@ -364,6 +364,16 @@ 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