snapshot...

modules/util/infer_cloud_fraction.py

@@ -6,11 +6,13 @@ from deeplearning.cloud_fraction_fcn_abi import run_evaluate_static, run_evaluat
 tf_model_name = model_path+'tf_model_cld_frac_abi_v1' + os.sep
 
 
-def infer_cloud_fraction(clvrx_dir, output_dir, model_path=model_path+tf_model_name):
+def infer_cloud_fraction(clvrx_dir, output_dir, model_path=model_path+tf_model_name, full_disk=True):
     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)
-        # run_evaluate_static_full_disk(fname, out_file, model_path)
+        if full_disk:
+            run_evaluate_static_full_disk(fname, out_file, model_path)
+        else:
+            run_evaluate_static(fname, out_file, model_path)