mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

Fix params format

Browse Source
This commit is contained in:
gaotingquan 2021-07-19 03:15:16 +00:00 committed by Tingquan Gao
parent a9f35981e9
commit a378af3886
2 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
deploy/hubserving/clas/params.py
View File

@ -33,8 +33,10 @@ def get_default_confg():
"enable_benchmark": False "enable_benchmark": False
}, },
'PostProcess': { 'PostProcess': {
'name': 'Topk', 'main_indicator': 'Topk',
'topk': 5, 'Topk': {
'class_id_map_file': './utils/imagenet1k_label_list.txt' 'topk': 5,
'class_id_map_file': './utils/imagenet1k_label_list.txt'
}
} }
} }

4
paddleclas.py
View File

@ -407,7 +407,7 @@ class PaddleClas(object):
Args: Args:
model_name: The model name supported by PaddleClas, default by None. If specified, override config. 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. 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. 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. topk: Return the top k prediction results with the highest score.
""" """
@ -459,7 +459,7 @@ class PaddleClas(object):
Args: Args:
input_data (str | NumPy.array): The path of image, or the directory containing images, or the URL of image from Internet. 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: Raises:
ImageTypeError: Illegal input_data. ImageTypeError: Illegal input_data.