diff --git a/modules/deeplearning/srcnn.py b/modules/deeplearning/srcnn.py index b5f6e704446959e35d0ffc173ead882a992df0d9..5f2146eac0b22ce8503aedde23604dd3cf89bb68 100644 --- a/modules/deeplearning/srcnn.py +++ b/modules/deeplearning/srcnn.py @@ -208,7 +208,7 @@ class SRCNN: label = data.copy() data = data[:, data_idx, 3:131:2, 3:131:2] - data = resample(x_64, y_64, data, t, s) + data = resample(y_64, x_64, data, s, t) data = np.expand_dims(data, axis=3) # label = label[:, label_idx, 3:131:2, 3:131:2] @@ -657,7 +657,7 @@ def run_evaluate_static(in_file, out_file, ckpt_dir): nda = np.load(in_file) nda = nda[:, data_idx, 3:131:2, 3:131:2] - nda = resample(x_64, y_64, nda, t, s) + nda = resample(y_64, x_64, nda, s, t) nda = np.expand_dims(nda, axis=3) nn = SRCNN()