From e153e3dbf8c4575602aa576ca1dd901a4b09d532 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 11 Apr 2023 11:46:31 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/cloud_fraction_fcn.py | 10 ++++++++++
1 file changed, 10 insertions(+)
diff --git a/modules/deeplearning/cloud_fraction_fcn.py b/modules/deeplearning/cloud_fraction_fcn.py
index 2feb826e..21afd777 100644
--- a/modules/deeplearning/cloud_fraction_fcn.py
+++ b/modules/deeplearning/cloud_fraction_fcn.py
@@ -8,6 +8,7 @@ import os, datetime
import numpy as np
import pickle
import h5py
+import xarray as xr
import gc
LOG_DEVICE_PLACEMENT = False
@@ -816,6 +817,15 @@ def run_evaluate_static(in_file, out_file, ckpt_dir):
bt = denormalize(bt, 'temp_11_0um_nom', mean_std_dct)
refl_avg = denormalize(refl_avg, 'refl_0_65um_nom', mean_std_dct)
+ var_names = ['cloud_fraction', 'temp_11_0um', 'refl_0_65um']
+ dims = ['num_params', 'y', 'x']
+
+ da = xr.DataArray(np.column_stack([cld_frac_out, bt, refl_avg]), dims=dims)
+ da.assign_coords({
+ 'lat': (['y', 'x'], lats),
+ 'lon': (['y', 'x'], lons)
+ })
+
if out_file is not None:
np.save(out_file, (cld_frac_out, bt, refl_avg, cp, lons, lats))
else:
--
GitLab