diff --git a/modules/deeplearning/esrgan_exp.py b/modules/deeplearning/esrgan_exp.py index 5fbd49f58bb99a5d12e7ec6f006ee9113d0a1c5c..dbdbfeab174c860242fa3248c1aa81e69a25e0a7 100644 --- a/modules/deeplearning/esrgan_exp.py +++ b/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,8 +123,8 @@ 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 = hr_image.numpy() hr_image = descale2(hr_image[:, :, :, 1], 0.0, 160.0) image = descale2(image[:, :, 1], 0.0, 160.0)