mirrors
/
PaddleOCR
mirror of https://github.com/PaddlePaddle/PaddleOCR.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix rec hpi config (#14905)

pull/15056/head
zhangyubo0722 2025-04-20 14:46:30 +08:00 committed by GitHub
parent b0ce52f729
commit 5d120f8fe9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ppocr/utils/export_model.py
View File

@ -49,7 +49,7 @@ def dump_infer_config(config, path, logger):
arch_config = config["Architecture"] arch_config = config["Architecture"]
if arch_config["algorithm"] in ["SVTR_LCNet", "SVTR_HGNet"]: if arch_config["algorithm"] in ["SVTR_LCNet", "SVTR_HGNet"]:
common_dynamic_shapes = { common_dynamic_shapes = {
"x": [[1, 3, 48, 160], [1, 3, 48, 320], [8, 3, 48, 640]] "x": [[1, 3, 48, 160], [1, 3, 48, 320], [8, 3, 48, 3200]]
} }
elif arch_config["model_type"] == "det": elif arch_config["model_type"] == "det":
common_dynamic_shapes = { common_dynamic_shapes = {