diff --git a/modules/util/infer_cloud_fraction.py b/modules/util/infer_cloud_fraction.py index 4f9bd9173b65b94bc3f37f5a8f4d45b39360c6fd..69e063e8b7bfd57ced9313ed93fd6d19f713871e 100644 --- a/modules/util/infer_cloud_fraction.py +++ b/modules/util/infer_cloud_fraction.py @@ -5,6 +5,7 @@ import os from deeplearning.cloud_fraction_fcn_abi import SRCNN from util.util import get_cartopy_crs, write_cld_frac_file_nc4 import numpy as np +import time def infer_cloud_fraction(clvrx_path, output_dir, full_disk=True, satellite='GOES16', domain='FD', pattern=None): @@ -33,6 +34,7 @@ def infer_cloud_fraction(clvrx_path, output_dir, full_disk=True, satellite='GOES clvrx_str_time = dto.strftime('%Y-%m-%d_%H:%M') out_file = output_dir + 'cloud_fraction_' + clvrx_str_time + '.nc' + t0 = time.time() if full_disk: cld_frac = nn.run_inference_full_disk(pname, None) else: @@ -40,4 +42,6 @@ def infer_cloud_fraction(clvrx_path, output_dir, full_disk=True, satellite='GOES 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) + t1 = time.time() + print('total time: ', (t1 - t0)) print('Done: ', pname)