snapshot...

modules/deeplearning/esrgan_exp.py

@@ -101,7 +101,7 @@ def preprocess_image(hr_image):
       hr_image = tmp_image
       hr_image *= 255.0
 
-  hr_image *= 255.0
+  # hr_image *= 255.0
 
   hr_size = (tf.convert_to_tensor(hr_image.shape[:-1]) // 4) * 4
   hr_image = tf.image.crop_to_bounding_box(hr_image, 0, 0, hr_size[0], hr_size[1])
@@ -123,11 +123,12 @@ def run(in_file, out_file):
     print('inference time: ', (t1-t0))
     sres_image = tf.squeeze(sres_image)
 
-    sres_image /= 255.0
-    hr_image /= 255.0
+    # sres_image /= 255.0
+    # hr_image /= 255.0
     hr_image = descale2(hr_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)
 
     if out_file is not None: