Fix params format
parent
a9f35981e9
commit
a378af3886
|
@ -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'
|
||||
}
|
||||
}
|
||||
}
|
|
@ -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.
|
||||
|
|
Loading…
Reference in New Issue