diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py index c506a4f563ed6ccdbed13e89810f39e70163dd25..12cc718586601be06b492b3033adeb7af82690e1 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)