PP-DocLayout-S / inference.yml
Tingquan's picture
Upload folder using huggingface_hub
d57ac8c verified
mode: paddle
draw_threshold: 0.5
metric: COCO
use_dynamic_shape: false
Global:
model_name: PP-DocLayout-S
arch: GFL
min_subgraph_size: 3
Preprocess:
- interp: 2
keep_ratio: false
target_size:
- 480
- 480
type: Resize
- is_scale: true
mean:
- 0.485
- 0.456
- 0.406
std:
- 0.229
- 0.224
- 0.225
type: NormalizeImage
- type: Permute
label_list:
- paragraph_title
- image
- text
- number
- abstract
- content
- figure_title
- formula
- table
- table_title
- reference
- doc_title
- footnote
- header
- algorithm
- footer
- seal
- chart_title
- chart
- formula_number
- header_image
- footer_image
- aside_text
Hpi:
backend_configs:
paddle_infer:
trt_dynamic_shapes: &id001
image:
- - 1
- 3
- 480
- 480
- - 1
- 3
- 480
- 480
- - 8
- 3
- 480
- 480
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