diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index 49f53adae0aa3f32c56945144cb56a41c1b73134..21ced214245eb0feedd55b59bcc265dede5568c1 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -718,12 +718,14 @@ def run_evaluate_static(in_file, out_file, ckpt_dir): grd_a = get_grid_values_all(h5f, 'temp_11_0um_nom') grd_a = grd_a[y_0:y_0+sub_y, x_0:x_0+sub_x] grd_a = grd_a[slc_y_2, slc_x_2] + bt = grd_a grd_a = normalize(grd_a, 'temp_11_0um_nom', mean_std_dct) grd_a = resample_2d_linear_one(x_2, y_2, grd_a, t, s) grd_b = get_grid_values_all(h5f, 'refl_0_65um_nom') grd_b = grd_b[y_0:y_0+sub_y, x_0:x_0+sub_x] grd_b = grd_b[slc_y_2, slc_x_2] + refl = grd_b grd_b = normalize(grd_b, 'refl_0_65um_nom', mean_std_dct) grd_b = resample_2d_linear_one(x_2, y_2, grd_b, t, s) @@ -744,7 +746,7 @@ def run_evaluate_static(in_file, out_file, ckpt_dir): if out_file is not None: np.save(out_file, out_sr) else: - return out_sr + return out_sr, bt, refl def run_evaluate_static_2(in_file, out_file, ckpt_dir):