From a378af38863dc538c1c552a2c25b18d09706f99c Mon Sep 17 00:00:00 2001 From: gaotingquan Date: Mon, 19 Jul 2021 03:15:16 +0000 Subject: [PATCH] Fix params format --- deploy/hubserving/clas/params.py | 10 ++++++---- paddleclas.py | 4 ++-- 2 files changed, 8 insertions(+), 6 deletions(-) diff --git a/deploy/hubserving/clas/params.py b/deploy/hubserving/clas/params.py index 358d116d..9d9c0dda 100644 --- a/deploy/hubserving/clas/params.py +++ b/deploy/hubserving/clas/params.py @@ -33,8 +33,10 @@ def get_default_confg(): "enable_benchmark": False }, 'PostProcess': { - 'name': 'Topk', - 'topk': 5, - 'class_id_map_file': './utils/imagenet1k_label_list.txt' + 'main_indicator': 'Topk', + 'Topk': { + 'topk': 5, + 'class_id_map_file': './utils/imagenet1k_label_list.txt' + } } - } \ No newline at end of file + } diff --git a/paddleclas.py b/paddleclas.py index 4c878691..72924a0f 100644 --- a/paddleclas.py +++ b/paddleclas.py @@ -407,7 +407,7 @@ class PaddleClas(object): Args: model_name: The model name supported by PaddleClas, default by None. If specified, override config. inference_model_dir: The directory that contained model file and params file to be used, default by None. If specified, override config. - use_gpu: Wheather use GPU, default by None. If specified, override config. + use_gpu: whether use GPU, default by None. If specified, override config. batch_size: The batch size to pridict, default by None. If specified, override config. topk: Return the top k prediction results with the highest score. """ @@ -459,7 +459,7 @@ class PaddleClas(object): Args: input_data (str | NumPy.array): The path of image, or the directory containing images, or the URL of image from Internet. - print_pred (bool, optional): Wheather print the prediction result. Defaults to False. + print_pred (bool, optional): whether print the prediction result. Defaults to False. Raises: ImageTypeError: Illegal input_data. -- GitLab