inference_cartoon.yaml 1.2 KB
Newer Older
B
Bin Lu 已提交
1
Global:
2 3 4
  infer_imgs: "./dataset/cartoon_demo_data_v1.0/query/"
  det_inference_model_dir: "./models/ppyolov2_r50vd_dcn_mainbody_v1.0_infer/"
  rec_inference_model_dir: "./models/cartoon_rec_ResNet50_iCartoon_v1.0_infer/"
B
Bin Lu 已提交
5 6
  batch_size: 1
  image_shape: [3, 640, 640]
littletomatodonkey's avatar
littletomatodonkey 已提交
7 8
  threshold: 0.2
  max_det_results: 5
B
Bin Lu 已提交
9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51
  labe_list:
  - foreground

  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

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: {}

DetPostProcess: {}

RecPreProcess:
  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:

RecPostProcess: null

IndexProcess:
52 53 54 55
  index_path: "./dataset/cartoon_demo_data_v1.0/index/"
  search_budget: 100
  return_k: 5
  dist_type: "IP"
littletomatodonkey's avatar
littletomatodonkey 已提交
56
  score_thres: 0.5