diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index e2dd4f4f6e381fbf80cd79d3565e7870a1a76900..20e1ae80014aca36932a4bd9ff701af6f0d44ad6 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -750,8 +750,10 @@ def run_evaluate_static(in_file, out_file, ckpt_dir): y_0, x_0, = 2432 - int(sub_y/2), 2432 - int(sub_x/2) h5f = h5py.File(in_file, 'r') + 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 = np.where(np.isnan(grd_a), 0, grd_a) hr_grd_a = grd_a.copy() hr_grd_a = hr_grd_a[y_128, x_128] grd_a = grd_a[slc_y_2, slc_x_2] @@ -768,6 +770,7 @@ def run_evaluate_static(in_file, out_file, ckpt_dir): grd_c = get_grid_values_all(h5f, label_param) # grd_c = gaussian_filter(grd_c, sigma=1.0) grd_c = grd_c[y_0:y_0+sub_y, x_0:x_0+sub_x] + grd_c = np.where(np.isnan(grd_c), 0, grd_c) hr_grd_c = grd_c.copy() hr_grd_c = hr_grd_c[y_128, x_128] grd_c = grd_c[slc_y_2, slc_x_2]