opensource/PaddleOCR
3
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2026-09-21 05:12:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix dy shapes of trt for rec models (#14654)

Browse Source
This commit is contained in:
Tingquan Gao
2025-02-11 11:17:23 +08:00
committed by GitHub
parent ff0509379c
commit 17fff8cca4
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ppocr/utils/export_model.py
+1 -1
View File
@@ -49,7 +49,7 @@ def dump_infer_config(config, path, logger):
arch_config = config["Architecture"]
if arch_config["algorithm"] in ["SVTR_LCNet", "SVTR_HGNet"]:
common_dynamic_shapes = {
"x": [[1, 3, 24, 160], [1, 3, 48, 320], [8, 3, 96, 640]]
"x": [[1, 3, 48, 160], [1, 3, 48, 320], [8, 3, 48, 640]]
}
elif arch_config["model_type"] == "det":
common_dynamic_shapes = {