From ad7780a63c23ee18599c90be0bd4d67e398b7387 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 6 Jun 2023 11:42:27 -0500
Subject: [PATCH] snapshot...

---
 modules/util/infer_cloud_fraction.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/modules/util/infer_cloud_fraction.py b/modules/util/infer_cloud_fraction.py
index 1e9262a0..71fb01cc 100644
--- a/modules/util/infer_cloud_fraction.py
+++ b/modules/util/infer_cloud_fraction.py
@@ -1,13 +1,19 @@
 from util.setup_cloud_fraction import model_path
+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
+output_dir = '/home/rink' + os.sep
 
 
-def infer_cloud_fraction(clavrx_file, model_path=model_path+tf_model_name):
-    cld_frac = run_evaluate_static(clavrx_file, None, model_path)
-    return cld_frac
+def infer_cloud_fraction(clvrx_dir, model_path=model_path+tf_model_name):
+    clvrx_ds = CLAVRx(clvrx_dir)
+    for fname 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
+        run_evaluate_static(fname, out_file, model_path)
 
 
 def infer_cloud_fraction_full_disk(clavrx_file, model_path=model_path+tf_model_name):
-- 
GitLab