diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index 86b39e4c235bdb57c4426e8b345c7fc99eeb0c75..ccaa9842e3bbe189d1b9bd26d96e2006771f325b 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -619,8 +619,8 @@ class ESPCN:
         ckpt.restore(ckpt_manager.latest_checkpoint)
 
         data = normalize(nda_lr, param, mean_std_dct)
-        data = np.expand_dims(data, axis=0)
-        data = np.expand_dims(data, axis=3)
+        #data = np.expand_dims(data, axis=0)
+        #data = np.expand_dims(data, axis=3)
 
         self.reset_test_metrics()