inference_cls.yaml

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"