inference_cartoon.yaml 1.2 KB
Newer Older
B
Bin Lu 已提交
1
Global:
2
  infer_imgs: "./recognition_demo_data_v1.0/test_cartoon"
3 4
  det_inference_model_dir: "./models/ppyolov2_r50vd_dcn_mainbody_v1.0_infer/"
  rec_inference_model_dir: "./models/cartoon_rec_ResNet50_iCartoon_v1.0_infer/"
D
dyning 已提交
5
  rec_nms_thresold: 0.05
6

B
Bin Lu 已提交
7 8
  batch_size: 1
  image_shape: [3, 640, 640]
littletomatodonkey's avatar
littletomatodonkey 已提交
9 10
  threshold: 0.2
  max_det_results: 5
B
Bin Lu 已提交
11 12 13 14
  labe_list:
  - foreground

  use_gpu: True
W
weishengyu 已提交
15
  enable_mkldnn: False
B
Bin Lu 已提交
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 52 53
  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:
54
  index_path: "./recognition_demo_data_v1.0/gallery_cartoon/index/"
55 56 57
  search_budget: 100
  return_k: 5
  dist_type: "IP"
littletomatodonkey's avatar
littletomatodonkey 已提交
58
  score_thres: 0.5