inference_det.yaml 734 字节
Newer Older
W
weishengyu 已提交
1 2 3 4 5 6 7
Global:
  infer_imgs: "./images/wangzai.jpg"
  det_inference_model_dir: "./models/ppyolov2_r50vd_dcn_mainbody_v1.0_infer"
  batch_size: 1
  image_shape: [3, 640, 640]
  threshold: 0.2
  max_det_results: 1
H
HydrogenSulfate 已提交
8
  label_list:
W
weishengyu 已提交
9 10 11 12
  - foreground

  # inference engine config
  use_gpu: True
13 14
  enable_mkldnn: True
  cpu_num_threads: 10
W
weishengyu 已提交
15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32
  enable_benchmark: True
  use_fp16: False
  ir_optim: True
  use_tensorrt: False
  gpu_mem: 8000
  enable_profile: False

DetPreProcess:
  transform_ops:
    - DetResize:
        interp: 2
        keep_ratio: false
        target_size: [640, 640]
    - DetNormalizeImage:
        is_scale: true
        mean: [0.485, 0.456, 0.406]
        std: [0.229, 0.224, 0.225]
    - DetPermute: {}
S
stephon 已提交
33
    
S
stephon 已提交
34
DetPostProcess: {}