diff --git a/modules/deeplearning/esrgan_exp.py b/modules/deeplearning/esrgan_exp.py
index d4e4436dedcc1ca950af711fc354f0227099d7fd..6a0db16376ec6a4286f78955eb071ece2a44913b 100644
--- a/modules/deeplearning/esrgan_exp.py
+++ b/modules/deeplearning/esrgan_exp.py
@@ -125,6 +125,7 @@ def run(in_file, out_file):
 
     # sres_image /= 255.0
     # 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)