diff --git a/deploy/hubserving/clas/params.py b/deploy/hubserving/clas/params.py index 358d116d3..9d9c0dda6 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 4c878691a..72924a0fb 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.