snapshot...

modules/util/infer_cloud_fraction.py

 from util.geos_nav import get_navigation
-from util.setup_cloud_fraction import model_path
+from util.setup_cloud_products import model_path_cld_frac, model_path_cld_opd
 from aeolus.datasource import CLAVRx
 import os
-from deeplearning.cloud_fraction_fcn_abi import SRCNN
+from deeplearning.cloud_fraction_fcn_abi import SRCNN as SRCNN_CLD_FRAC
+from deeplearning.cloud_opd_fcn_abi import SRCNN as SRCNN_CLD_OPD
 from util.util import get_cartopy_crs, write_cld_frac_file_nc4
 import numpy as np
 import time
@@ -10,8 +11,11 @@ import time
 
 def infer_cloud_fraction(clvrx_path, output_dir, full_disk=True, satellite='GOES16', domain='FD', pattern=None):
     # -- location of the trained model
-    ckpt_dir_s = os.listdir(model_path)
-    ckpt_dir = model_path + ckpt_dir_s[0]
+    ckpt_dir_s = os.listdir(model_path_cld_frac)
+    ckpt_dir_cld_frac = model_path_cld_frac + ckpt_dir_s[0]
+
+    ckpt_dir_s = os.listdir(model_path_cld_opd)
+    ckpt_dir_cld_opd = model_path_cld_opd + ckpt_dir_s[0]
 
     # -- Navigation parameters
     geos, xlen, xmin, xmax, ylen, ymin, ymax = get_cartopy_crs(satellite, domain)
@@ -22,8 +26,11 @@ def infer_cloud_fraction(clvrx_path, output_dir, full_disk=True, satellite='GOES
     y_rad = ll * nav.LFAC + nav.LOFF
 
     # -- Create a model instance and initialize with trained model above
-    nn = SRCNN()
-    nn.setup_inference(ckpt_dir)
+    nn_cld_frac = SRCNN_CLD_FRAC()
+    nn_cld_frac.setup_inference(ckpt_dir_cld_frac)
+
+    nn_cld_opd = SRCNN_CLD_OPD()
+    nn_cld_opd.setup_inference(ckpt_dir_cld_opd)
 
     if pattern is not None:
         clvrx_ds = CLAVRx(clvrx_path, pattern=pattern)
@@ -36,9 +43,11 @@ def infer_cloud_fraction(clvrx_path, output_dir, full_disk=True, satellite='GOES
 
         t0 = time.time()
         if full_disk:
-            cld_frac = nn.run_inference_full_disk(pname, None)
+            cld_frac = nn_cld_frac.run_inference_full_disk(pname, None)
+            cld_opd = nn_cld_opd.run_inference_full_disk(pname, None)
         else:
-            cld_frac = nn.run_inference(pname, None)
+            cld_frac = nn_cld_frac.run_inference(pname, None)
+            cld_opd = nn_cld_opd.run_inference(pname, 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)