{ | |
"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 | |
] | |
} |