diff --git a/deploy/configs/inference_cls.yaml b/deploy/configs/inference_cls.yaml new file mode 100644 index 000000000..4b15f3f4e --- /dev/null +++ b/deploy/configs/inference_cls.yaml @@ -0,0 +1,30 @@ + +Global: + infer_imgs: "../docs/images/whl/demo.jpg" + inference_model_dir: "./MobileNetV1_infer/" + batch_size: 1 + use_gpu: True + enable_mkldnn: True + 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 + class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" \ No newline at end of file