diff --git a/modules/util/infer_cloud_fraction.py b/modules/util/infer_cloud_fraction.py index c7f51f6ca319f83488f90fd0b7f6f30c00ed1ac3..1546545b19e033f4338454d4fdeb38515345bb85 100644 --- a/modules/util/infer_cloud_fraction.py +++ b/modules/util/infer_cloud_fraction.py @@ -1,10 +1,13 @@ from util.setup_cloud_fraction import model_path +import os from deeplearning.cloud_fraction_fcn_abi import run_evaluate_static, run_evaluate_static_full_disk +tf_model_name = model_path+'tf_model_cld_frac_abi_v1' + os.sep -def infer_cloud_fraction(clavrx_file, model_path=model_path): + +def infer_cloud_fraction(clavrx_file, model_path=model_path+tf_model_name): cld_frac = run_evaluate_static(clavrx_file, None, model_path) -def infer_cloud_fraction_full_disk(clavrx_file, model_path=model_path): +def infer_cloud_fraction_full_disk(clavrx_file, model_path=model_path+tf_model_name): cld_frac = run_evaluate_static_full_disk(clavrx_file, None, model_path) \ No newline at end of file