From ee577979ce1177fbc287dc5c239249437b358c9e Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Fri, 12 Aug 2022 14:07:23 -0500 Subject: [PATCH] test... --- modules/deeplearning/espcn.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py index c506a4f5..12cc7185 100644 --- a/modules/deeplearning/espcn.py +++ b/modules/deeplearning/espcn.py @@ -195,14 +195,16 @@ class ESPCN: label = np.concatenate(label_s) # label = label[:, label_idx, :, :] - label = label[:, label_idx, 3:67, 3:67] + #label = label[:, label_idx, 3:67, 3:67] + label = label[:, label_idx, 0:32, 0:32] label = np.expand_dims(label, axis=3) - label = tf.image.resize(label, (32, 32), method='nearest').numpy() + #label = tf.image.resize(label, (32, 32), method='nearest').numpy() # data = data[:, data_idx, :, :] - data = data[:, data_idx, 3:67, 3:67] + #data = data[:, data_idx, 3:67, 3:67] + data = data[:, data_idx, 0:32, 0:32] data = np.expand_dims(data, axis=3) - data = tf.image.resize(data, (32, 32), method='nearest').numpy() + #data = tf.image.resize(data, (32, 32), method='nearest').numpy() # data = tf.image.resize(data, (36, 36)).numpy() data = data.astype(np.float32) -- GitLab