snapshot...

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)