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

snapshot...

parent 59714a2e
No related merge requests found
......@@ -406,15 +406,15 @@ 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])
# 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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment