snapshot...

modules/deeplearning/esrgan_exp.py

@@ -6,7 +6,7 @@ import tensorflow as tf
 #import tensorflow_hub as hub
 import matplotlib.pyplot as plt
 import h5py
-from util.util import scale, descale, get_grid_values_all
+from util.util import descale2, get_grid_values_all
 from util.setup import home_dir
 
 target_param = 'cld_opd_dcomp'
@@ -123,6 +123,13 @@ 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
+    hr_image = descale2(hr_image[:, :, :, 1], 0.0, 160.0)
+    image = descale2(image[:, :, :, 1], 0.0, 160.0)
+
+    sres_image = descale2(sres_image, 0.0, 160.0)
+
     if out_file is not None:
         # np.save(out_file, sres_image)
         with open(out_file, 'wb') as f: