2021-06-04 09:29:01 +00:00
|
|
|
Global:
|
2021-06-18 17:44:04 +08:00
|
|
|
infer_imgs: "./images/ILSVRC2012_val_00000010.jpeg"
|
|
|
|
|
inference_model_dir: "./models"
|
2021-06-04 09:29:01 +00:00
|
|
|
batch_size: 1
|
|
|
|
|
use_gpu: True
|
2021-06-19 02:10:53 +08:00
|
|
|
enable_mkldnn: False
|
2021-06-04 09:29:01 +00:00
|
|
|
cpu_num_threads: 100
|
|
|
|
|
enable_benchmark: True
|
|
|
|
|
use_fp16: False
|
|
|
|
|
ir_optim: True
|
|
|
|
|
use_tensorrt: False
|
|
|
|
|
gpu_mem: 8000
|
|
|
|
|
enable_profile: False
|
|
|
|
|
PreProcess:
|
|
|
|
|
transform_ops:
|
|
|
|
|
- ResizeImage:
|
|
|
|
|
resize_short: 256
|
|
|
|
|
- CropImage:
|
|
|
|
|
size: 224
|
|
|
|
|
- NormalizeImage:
|
|
|
|
|
scale: 0.00392157
|
|
|
|
|
mean: [0.485, 0.456, 0.406]
|
|
|
|
|
std: [0.229, 0.224, 0.225]
|
|
|
|
|
order: ''
|
|
|
|
|
- ToCHWImage:
|
|
|
|
|
PostProcess:
|
|
|
|
|
name: Topk
|
|
|
|
|
topk: 5
|
2021-06-13 05:02:50 +00:00
|
|
|
class_id_map_file: "../ppcls/utils/imagenet1k_label_list.txt"
|
