Fix params format

pull/1059/head
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

View File

@ -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'
}
}
}

View File

@ -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.