cpp_demo.yml 575 字节
Newer Older
W
wangguanzhong 已提交
1
# demo for cpp_infer.py
W
wangguanzhong 已提交
2 3 4

mode: trt_fp32 # trt_fp32, trt_fp16, trt_int8, fluid
arch: RCNN # YOLO, SSD, RCNN, RetinaNet
5
min_subgraph_size: 40 # need 3 for YOLO arch
W
wangguanzhong 已提交
6 7
use_python_inference: False # whether to use python inference

W
wangguanzhong 已提交
8
# visualize the predicted image
W
wangguanzhong 已提交
9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27
metric: COCO # COCO, VOC
draw_threshold: 0.5

Preprocess:
- type: Resize
  target_size: 640
  max_size: 640
- type: Normalize
  mean:
  - 0.485
  - 0.456
  - 0.406
  std:
  - 0.229
  - 0.224
  - 0.225
  is_scale: True
- type: Permute
  to_bgr: False
28 29
- type: PadStride
  stride: 0 # set 32 on FPN and 128 on RetinaNet