From 0805f3610e3a4fdf121c4f3b24d0c108e5ccb987 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 12 Jun 2023 15:32:36 -0500
Subject: [PATCH] snapshot...
---
modules/util/infer_cloud_fraction.py | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/modules/util/infer_cloud_fraction.py b/modules/util/infer_cloud_fraction.py
index 543e30be..de1c1af3 100644
--- a/modules/util/infer_cloud_fraction.py
+++ b/modules/util/infer_cloud_fraction.py
@@ -3,16 +3,17 @@ from aeolus.datasource import CLAVRx
import os
from deeplearning.cloud_fraction_fcn_abi import run_evaluate_static, run_evaluate_static_full_disk
-tf_model_name = model_path+'tf_model_cld_frac_abi_v1' + os.sep
+def infer_cloud_fraction(clvrx_path, output_dir, full_disk=True):
+ ckpt_dir_s = os.listdir(model_path)
+ ckpt_dir = model_path + ckpt_dir_s[0]
-def infer_cloud_fraction(clvrx_path, output_dir, model_path=model_path+tf_model_name, full_disk=True):
clvrx_ds = CLAVRx(clvrx_path)
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
if full_disk:
- run_evaluate_static_full_disk(fname, out_file, model_path)
+ run_evaluate_static_full_disk(fname, out_file, ckpt_dir)
else:
- run_evaluate_static(fname, out_file, model_path)
+ run_evaluate_static(fname, out_file, ckpt_dir)
--
GitLab