diff --git a/modules/deeplearning/cloud_fraction_fcn_abi.py b/modules/deeplearning/cloud_fraction_fcn_abi.py index 00cd3bad86947145be34485eeaed75fc42e1ee4a..fad30b296ef4f8bbb8e615017ed543f67a407f6e 100644 --- a/modules/deeplearning/cloud_fraction_fcn_abi.py +++ b/modules/deeplearning/cloud_fraction_fcn_abi.py @@ -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