diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index 40a79ee04f43d00fe6ddbb50ab0681352ec0f9b0..71ce8bfb212ca153f53063b9cb1f4ed29d6d7a90 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -196,7 +196,7 @@ 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))
+        label = tf.image.resize(label, (32, 32)).numpy()
 
         # data = data[:, data_idx, :, :]
         data = data[:, data_idx, 3:67, 3:67]