diff --git a/modules/deeplearning/cloud_opd_fcn_abi.py b/modules/deeplearning/cloud_opd_fcn_abi.py index 270dc1fa4cc06a2d9866757ec23a642eaf9fedb2..d4ce79a6fe7adaf019f3794eeb68055caae9005e 100644 --- a/modules/deeplearning/cloud_opd_fcn_abi.py +++ b/modules/deeplearning/cloud_opd_fcn_abi.py @@ -888,7 +888,8 @@ def run_restore_static(directory, ckpt_dir, out_file=None): if out_file is not None: y_hi, x_hi = (Y_LEN // 4) + 1, (X_LEN // 4) + 1 np.save(out_file, - [labels[:, :, :, 0], preds[:, :, :, 0], + [labels[:, :, :, 0], + preds[:, :, :, 0], inputs[:, 1:y_hi, 1:x_hi, 0], descale(inputs[:, 1:y_hi, 1:x_hi, 1], 'refl_0_65um_nom', mean_std_dct), descale(inputs[:, 1:y_hi, 1:x_hi, 2], 'refl_0_65um_nom', mean_std_dct),