diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index 9b89fc7ebac92220ac310d90a721261faaec9ed9..e855896833b3eb500005d1bf9e7f09042df1d4c0 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -789,6 +789,7 @@ def run_evaluate_static(in_file, out_file, ckpt_dir): #data = np.stack([grd_a, grd_c], axis=2) data = np.stack([grd_c], axis=2) data = np.expand_dims(data, axis=0) + data = tf.convert_to_tensor(data, dtype=tf.float32) nn = SRCNN() out_sr = nn.run_evaluate(data, ckpt_dir)