ppocr-v5-rec.yaml 420 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
# config/ppocr-v5-rec.yaml
model:
  name: "ppocr-v5-rec"
  
  # 模型文件路径
  model_file: "/data/wkx/benchmark/models/ppocr-v5-rec_model.onnx"
  
  # 测试的batch大小
  batch_sizes: [1, 8]
  
  # 输入配置
  inputs:
    - name: "x"
      shape: [3, 48, 320]  # BATCH会在运行时插入到第一个位置
  
  # 环境变量
  env_vars:
    # MIGRAPHX_ENABLE_NHWC: 1
    
  # 额外参数
  extra_args: []