PicoDet_layout_1x_table / inference.yml
Tingquan's picture
Upload folder using huggingface_hub
a4def94 verified
mode: paddle
draw_threshold: 0.5
metric: COCO
use_dynamic_shape: false
Global:
model_name: PicoDet_layout_1x_table
arch: GFL
min_subgraph_size: 3
Preprocess:
- interp: 2
keep_ratio: false
target_size:
- 800
- 608
type: Resize
- is_scale: true
mean:
- 0.485
- 0.456
- 0.406
std:
- 0.229
- 0.224
- 0.225
type: NormalizeImage
- type: Permute
- stride: 32
type: PadStride
label_list:
- Table
Hpi:
backend_configs:
paddle_infer:
trt_dynamic_shapes: &id001
image:
- - 1
- 3
- 800
- 608
- - 1
- 3
- 800
- 608
- - 8
- 3
- 800
- 608
scale_factor:
- - 1
- 2
- - 1
- 2
- - 8
- 2
trt_dynamic_shape_input_data:
scale_factor:
- - 2
- 2
- - 1
- 1
- - 0.67
- 0.67
- 0.67
- 0.67
- 0.67
- 0.67
- 0.67
- 0.67
- 0.67
- 0.67
- 0.67
- 0.67
- 0.67
- 0.67
- 0.67
- 0.67
tensorrt:
dynamic_shapes: *id001
NMS:
keep_top_k: 100
name: MultiClassNMS
nms_threshold: 0.5
nms_top_k: 1000
score_threshold: 0.3
fpn_stride:
- 8
- 16
- 32
- 64