inference.yml
1.4 KB · 101 lines · yaml Raw
1 mode: paddle
2 draw_threshold: 0.5
3 metric: COCO
4 use_dynamic_shape: false
5 Global:
6 model_name: PP-DocLayoutV3
7 arch: DETR
8 min_subgraph_size: 3
9 Preprocess:
10 - interp: 2
11 keep_ratio: false
12 target_size:
13 - 800
14 - 800
15 type: Resize
16 - mean:
17 - 0.0
18 - 0.0
19 - 0.0
20 norm_type: none
21 std:
22 - 1.0
23 - 1.0
24 - 1.0
25 type: NormalizeImage
26 - type: Permute
27 label_list:
28 - abstract
29 - algorithm
30 - aside_text
31 - chart
32 - content
33 - display_formula
34 - doc_title
35 - figure_title
36 - footer
37 - footer_image
38 - footnote
39 - formula_number
40 - header
41 - header_image
42 - image
43 - inline_formula
44 - number
45 - paragraph_title
46 - reference
47 - reference_content
48 - seal
49 - table
50 - text
51 - vertical_text
52 - vision_footnote
53 Hpi:
54 backend_configs:
55 paddle_infer:
56 trt_dynamic_shapes: &id001
57 image:
58 - - 1
59 - 3
60 - 800
61 - 800
62 - - 1
63 - 3
64 - 800
65 - 800
66 - - 8
67 - 3
68 - 800
69 - 800
70 scale_factor:
71 - - 1
72 - 2
73 - - 1
74 - 2
75 - - 8
76 - 2
77 trt_dynamic_shape_input_data:
78 scale_factor:
79 - - 2
80 - 2
81 - - 1
82 - 1
83 - - 0.67
84 - 0.67
85 - 0.67
86 - 0.67
87 - 0.67
88 - 0.67
89 - 0.67
90 - 0.67
91 - 0.67
92 - 0.67
93 - 0.67
94 - 0.67
95 - 0.67
96 - 0.67
97 - 0.67
98 - 0.67
99 tensorrt:
100 dynamic_shapes: *id001
101