From 5b7d235052da3de54ce95a8f0d90056d0f9b3b93 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Mon, 9 Oct 2023 16:22:36 -0500 Subject: [PATCH] snapshot... --- modules/util/infer_cloud_products.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/modules/util/infer_cloud_products.py b/modules/util/infer_cloud_products.py index 7cbad5af..d34ac002 100644 --- a/modules/util/infer_cloud_products.py +++ b/modules/util/infer_cloud_products.py @@ -44,10 +44,15 @@ def infer_cloud_products(clvrx_path, output_dir, full_disk=True, satellite='GOES t0 = time.time() if full_disk: cld_frac = nn_cld_frac.run_inference_full_disk(pname, None) - cld_opd = nn_cld_opd.run_inference_full_disk(pname, None) + cld_opd, opd = nn_cld_opd.run_inference_full_disk(pname, None) else: cld_frac = nn_cld_frac.run_inference(pname, None) - cld_opd = nn_cld_opd.run_inference(pname, None) + cld_opd, opd = nn_cld_opd.run_inference(pname, None) + + clr_cat = cld_frac == 0 + cld_cat = cld_frac == 4 + cld_opd[clr_cat] = opd[clr_cat] + cld_opd[cld_cat] = opd[cld_cat] write_cld_prods_file_nc4(clvrx_str_time, out_file, cld_frac, cld_opd, x_rad, y_rad, None, None, satellite=satellite, domain=domain, has_time=True) -- GitLab