ppocr-v5-det.yaml 700 Bytes
Newer Older
wangkx1's avatar
init  
wangkx1 committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
# config/ppocr-v5-det.yaml
model:
  name: "ppocr-v5-det"
  
  # 多个模型文件(对应不同batch size)
  model_files:
    - path: "/data/wkx/models/ppocr-v5-det_model-new-1x3x640x640.onnx"
      batch: 1
    # - path: "../models/yolov3/yolov3_b4.onnx"
    #   batch: 4
    - path: "/data/wkx/benchmark/models/ppocr-v5-det_model.onnx"
      batch: 8
    # - path: "../models/yolov3/yolov3_b16.onnx"
    #   batch: 16
  
  # 测试的batch大小(如果使用model_files,这里也可以指定)
  batch_sizes: [1, 8]
  
  # 输入配置
  inputs:
    - name: "x"
      shape: [3, 640, 640]
  
  # 环境变量
  env_vars:
    # MIGRAPHX_ENABLE_NHWC: 1
    
  # 额外参数
  extra_args: []