diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index ccaa9842e3bbe189d1b9bd26d96e2006771f325b..d9df97b25f9d2df21bd3eae10e2469f18a8cb5b1 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -209,7 +209,8 @@ class ESPCN:
         data = np.expand_dims(data, axis=3)
 
         # label = label[:, label_idx, :, :]
-        label = label[:, label_idx, 3:131:2, 3:131:2]
+        # label = label[:, label_idx, 3:131:2, 3:131:2]
+        label = label[:, label_idx, 3:131, 3:131]
         # label = label[:, label_idx, 3:67, 3:67]
         label = np.expand_dims(label, axis=3)
 
@@ -387,10 +388,10 @@ class ESPCN:
         conv = conv + conv_b
         print(conv.shape)
 
-        # conv = tf.keras.layers.Conv2D(num_filters * (factor ** 2), 3, padding='same')(conv)
+        conv = tf.keras.layers.Conv2D(num_filters * (factor ** 2), 3, padding='same')(conv)
         print(conv.shape)
 
-        #conv = tf.nn.depth_to_space(conv, factor)
+        conv = tf.nn.depth_to_space(conv, factor)
         #conv = tf.keras.layers.Conv2DTranspose(num_filters * (factor ** 2), 3, padding='same')(conv)
         print(conv.shape)