diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py index c9bd3abdc6943cae106f25aa62a5b86f40974b16..dc11ac894ae8f499f06279158afbbc96b275f673 100644 --- a/modules/deeplearning/espcn.py +++ b/modules/deeplearning/espcn.py @@ -196,12 +196,12 @@ class ESPCN: # label = label[:, label_idx, :, :] label = label[:, label_idx, 3:67, 3:67] 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 = 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)