diff --git a/modules/deeplearning/cloud_opd_fcn_abi.py b/modules/deeplearning/cloud_opd_fcn_abi.py index b23145974702a9bd8eca00e00354cb9f729f5891..1e6a4f93436386d5b1ff6d493fce9177d3a68b48 100644 --- a/modules/deeplearning/cloud_opd_fcn_abi.py +++ b/modules/deeplearning/cloud_opd_fcn_abi.py @@ -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):