diff --git a/modules/deeplearning/esrgan_exp.py b/modules/deeplearning/esrgan_exp.py
index 6a0db16376ec6a4286f78955eb071ece2a44913b..5fbd49f58bb99a5d12e7ec6f006ee9113d0a1c5c 100644
--- a/modules/deeplearning/esrgan_exp.py
+++ b/modules/deeplearning/esrgan_exp.py
@@ -127,7 +127,7 @@ def run(in_file, out_file):
     # hr_image /= 255.0
     hr_image = hr_image.numpy()
     hr_image = descale2(hr_image[:, :, :, 1], 0.0, 160.0)
-    image = descale2(image[:, :, :, 1], 0.0, 160.0)
+    image = descale2(image[:, :, 1], 0.0, 160.0)
 
     sres_image = sres_image.numpy()
     sres_image = descale2(sres_image, 0.0, 160.0)