fix rec hpi config ()

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
ppocr/utils

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, 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":
common_dynamic_shapes = {