yujunjun/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2026-01-08 04:56:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix bs to 1 in trt dy shape config for some formule rec model and tabel rec models (#14807)

Browse Source
This commit is contained in:
zhangyubo0722 2025-03-06 11:34:08 +08:00 committed by GitHub
parent 534e66506f
commit f95280d52d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
ppocr/utils/export_model.py
View File

@ -56,16 +56,21 @@ def dump_infer_config(config, path, logger):
"x": [[1, 3, 160, 160], [1, 3, 640, 640], [1, 3, 1280, 1280]]
}
elif arch_config["algorithm"] == "SLANet":
common_dynamic_shapes = {
"x": [[1, 3, 32, 32], [1, 3, 64, 448], [8, 3, 488, 488]]
}
if config["Global"].get("pdx_model_name", None) == "SLANet_plus":
common_dynamic_shapes = {
"x": [[1, 3, 32, 32], [1, 3, 64, 448], [1, 3, 488, 488]]
}
else:
common_dynamic_shapes = {
"x": [[1, 3, 32, 32], [1, 3, 64, 448], [8, 3, 488, 488]]
}
elif arch_config["algorithm"] == "SLANeXt":
common_dynamic_shapes = {
"x": [[1, 3, 512, 512], [1, 3, 512, 512], [8, 3, 512, 512]]
"x": [[1, 3, 512, 512], [1, 3, 512, 512], [1, 3, 512, 512]]
}
elif arch_config["algorithm"] == "LaTeXOCR":
common_dynamic_shapes = {
"x": [[1, 1, 32, 32], [1, 1, 64, 448], [8, 1, 192, 672]]
"x": [[1, 1, 32, 32], [1, 1, 64, 448], [1, 1, 192, 672]]
}
elif arch_config["algorithm"] == "UniMERNet":
common_dynamic_shapes = {