diff --git a/modules/deeplearning/cloud_opd_srcnn_abi.py b/modules/deeplearning/cloud_opd_srcnn_abi.py index 75720812c106edd2d5d3ff881d179e32ab2ddf04..1a7d51b0e7ea9105423765ed923e13dfcd77e4c0 100644 --- a/modules/deeplearning/cloud_opd_srcnn_abi.py +++ b/modules/deeplearning/cloud_opd_srcnn_abi.py @@ -758,6 +758,7 @@ def run_evaluate_static(in_file, out_file, ckpt_dir): print('OPD done') data = np.stack([bt_us, refl_us, refl_sub_lo, refl_sub_hi, refl_sub_std, cld_opd_us], axis=3) + print('INPUT: ', data.shape) cld_opd_sres = nn.run_evaluate(data, ckpt_dir) cld_opd_sres = descale(cld_opd_sres, label_param, mean_std_dct) @@ -780,7 +781,7 @@ def run_evaluate_static(in_file, out_file, ckpt_dir): if out_file is not None: # np.save(out_file, (cld_opd_sres_out, refl_out, cld_opd_out, cld_opd_hres)) - np.save(out_file, (cld_opd_sres_out, refl_us, cld_opd_us, cld_opd_us)) + np.save(out_file, cld_opd_sres_out) else: return cld_opd_sres_out, bt, refl