test_cls_connector.yml 301 字节
Newer Older
W
wangguanzhong 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
ENV:
  run_mode: paddle
  device: GPU
  min_subgraph_size: 3
  trt_calib_mode: False
  cpu_threads: 1
  trt_use_static: False

MODEL:
  - ClsCorrectionOp:
      name: cls_corr
      class_num: 4
      threshold: 0.9
      Inputs:
        - input.image
        - input.class_ids
        - input.scores