diff --git a/modules/deeplearning/cloud_opd_srcnn_abi_v2.py b/modules/deeplearning/cloud_opd_srcnn_abi_v2.py index eb443831589cb8cfc7f67a8b732f7a1eeb152433..943aa0f565a83dd99a6bec8dd84aacf73fcfa85e 100644 --- a/modules/deeplearning/cloud_opd_srcnn_abi_v2.py +++ b/modules/deeplearning/cloud_opd_srcnn_abi_v2.py @@ -329,6 +329,9 @@ class SRCNN: data_lr = np.flip(data, axis=2) label_lr = np.flip(label, axis=2) + # data_rot = np.rot90(data, axes=(1, 2)) + # label_rot = np.rot90(label, axes=(1, 2)) + data = np.concatenate([data, data_ud, data_lr]) label = np.concatenate([label, label_ud, label_lr])