PP-DocLayout-M / config.json
Tingquan's picture
Upload folder using huggingface_hub
7b89cad verified
{
"mode": "paddle",
"draw_threshold": 0.5,
"metric": "COCO",
"use_dynamic_shape": false,
"Global": {
"model_name": "PP-DocLayout-M"
},
"arch": "GFL",
"min_subgraph_size": 3,
"Preprocess": [
{
"interp": 2,
"keep_ratio": false,
"target_size": [
640,
640
],
"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": {
"image": [
[
1,
3,
640,
640
],
[
1,
3,
640,
640
],
[
8,
3,
640,
640
]
],
"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": {
"image": [
[
1,
3,
640,
640
],
[
1,
3,
640,
640
],
[
8,
3,
640,
640
]
],
"scale_factor": [
[
1,
2
],
[
1,
2
],
[
8,
2
]
]
}
}
}
},
"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
]
}