Skip to content
Snippets Groups Projects
Commit 5eadd3c9 authored by tomrink's avatar tomrink
Browse files

snapshot...

parent 84fca429
No related branches found
No related tags found
No related merge requests found
from util.geos_nav import get_navigation
from util.setup_cloud_fraction import model_path
from aeolus.datasource import CLAVRx
import os
from deeplearning.cloud_fraction_fcn_abi import SRCNN, run_evaluate_static, run_evaluate_static_full_disk
from util.util import get_cartopy_crs, write_cld_frac_file_nc4
import numpy as np
def infer_cloud_fraction(clvrx_path, output_dir, full_disk=True):
......@@ -19,10 +22,20 @@ def infer_cloud_fraction(clvrx_path, output_dir, full_disk=True):
run_evaluate_static(fname, out_file, ckpt_dir)
def infer_cloud_fraction_new(clvrx_path, output_dir, full_disk=True):
def infer_cloud_fraction_new(clvrx_path, output_dir, full_disk=True, satellite='GOES16', domain='FD'):
# location of the trained model
ckpt_dir_s = os.listdir(model_path)
ckpt_dir = model_path + ckpt_dir_s[0]
# Navigation parameters
geos, xlen, xmin, xmax, ylen, ymin, ymax = get_cartopy_crs(satellite, domain)
nav = get_navigation(satellite, domain)
cc = np.arange(xlen)
ll = np.arange(ylen)
x_rad = cc * nav.CFAC + nav.COFF
y_rad = ll * nav.LFAC + nav.LOFF
# Create a model instance and initialize with trained model above
nn = SRCNN()
nn.setup_inference(ckpt_dir)
......@@ -30,8 +43,11 @@ def infer_cloud_fraction_new(clvrx_path, output_dir, full_disk=True):
for fname, t_start, t_stop in clvrx_ds:
dto = clvrx_ds.get_datetime(fname)
clvrx_str_time = dto.strftime('%Y-%m-%d_%H:%M')
out_file = output_dir + 'cloud_fraction_' + clvrx_str_time
out_file = output_dir + 'cloud_fraction_' + clvrx_str_time + '.nc'
if full_disk:
nn.run_inference_full_disk(fname, out_file)
cld_frac = nn.run_inference_full_disk(fname, None)
else:
nn.run_inference(fname, out_file)
\ No newline at end of file
cld_frac = nn.run_inference(fname, None)
write_cld_frac_file_nc4(clvrx_str_time, out_file, cld_frac, x_rad, y_rad, None, None, satellite=satellite,
domain=domain, has_time=True)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment