fix: fix path

pull/1987/head
gaotingquan 2022-06-08 10:14:38 +00:00
parent f3a7cd42a7
commit 3a3b11ae9b
No known key found for this signature in database
GPG Key ID: F3EF7F42536A30B7
2 changed files with 2 additions and 1 deletions

View File

@ -1,7 +1,7 @@
include LICENSE.txt
include README.md
include docs/en/whl_en.md
recursive-include deploy/python predict_cls.py preprocess.py postprocess.py det_preprocess.py
recursive-include deploy/python *.py
recursive-include deploy/configs *.yaml
recursive-include deploy/utils get_image_list.py config.py logger.py predictor.py

View File

@ -194,6 +194,7 @@ class InputModelError(Exception):
def init_config(model_type, model_name, inference_model_dir, **kwargs):
cfg_path = f"deploy/configs/PULC/{model_name}/inference_{model_name}.yaml" if model_type == "pulc" else "deploy/configs/inference_cls.yaml"
cfg_path = os.path.join(__dir__, cfg_path)
cfg = config.get_config(cfg_path, show=False)
cfg.Global.inference_model_dir = inference_model_dir