提交 a378af38 编写于 作者: G gaotingquan 提交者: Tingquan Gao

Fix params format

上级 a9f35981
......@@ -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
}
......@@ -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.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册