diff --git a/modules/util/infer_cloud_fraction.py b/modules/util/infer_cloud_fraction.py index fe40f96aca3ac82bc9eb3a9d59d4697231600fb9..be92f61abd00ed6af1e889f0c85b44b74c14c81d 100644 --- a/modules/util/infer_cloud_fraction.py +++ b/modules/util/infer_cloud_fraction.py @@ -8,11 +8,11 @@ import numpy as np def infer_cloud_fraction(clvrx_path, output_dir, full_disk=True, satellite='GOES16', domain='FD'): - # location of the trained model + # -- location of the trained model ckpt_dir_s = os.listdir(model_path) ckpt_dir = model_path + ckpt_dir_s[0] - # Navigation parameters + # -- Navigation parameters geos, xlen, xmin, xmax, ylen, ymin, ymax = get_cartopy_crs(satellite, domain) nav = get_navigation(satellite, domain) cc = np.arange(xlen) @@ -20,7 +20,7 @@ def infer_cloud_fraction(clvrx_path, output_dir, full_disk=True, satellite='GOES x_rad = cc * nav.CFAC + nav.COFF y_rad = ll * nav.LFAC + nav.LOFF - # Create a model instance and initialize with trained model above + # -- Create a model instance and initialize with trained model above nn = SRCNN() nn.setup_inference(ckpt_dir)