opensource/PaddleOCR
3
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2026-08-30 17:23:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix formula rec models hpi_config (#14739)

Browse Source 
1. for formula rec models, the channel of input data is 1;
2. for latex_ocr_rec models, fix min/max size of dynamic shape.
This commit is contained in:
zhangyubo0722
2025-02-25 14:59:49 +08:00
committed by GitHub
parent 7d0147d752
commit 2b7b76310b
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ppocr/utils/export_model.py
+4 -4
View File
@@ -65,19 +65,19 @@ def dump_infer_config(config, path, logger):
}
elif arch_config["algorithm"] == "LaTeXOCR":
common_dynamic_shapes = {
"x": [[1, 3, 224, 224], [1, 3, 448, 448], [8, 3, 1280, 1280]]
"x": [[1, 1, 32, 32], [1, 1, 64, 448], [8, 1, 192, 672]]
}
elif arch_config["algorithm"] == "UniMERNet":
common_dynamic_shapes = {
"x": [[1, 3, 192, 672], [1, 3, 192, 672], [8, 3, 192, 672]]
"x": [[1, 1, 192, 672], [1, 1, 192, 672], [8, 1, 192, 672]]
}
elif arch_config["algorithm"] == "PP-FormulaNet-L":
common_dynamic_shapes = {
"x": [[1, 3, 768, 768], [1, 3, 768, 768], [8, 3, 768, 768]]
"x": [[1, 1, 768, 768], [1, 1, 768, 768], [8, 1, 768, 768]]
}
elif arch_config["algorithm"] == "PP-FormulaNet-S":
common_dynamic_shapes = {
"x": [[1, 3, 384, 384], [1, 3, 384, 384], [8, 3, 384, 384]]
"x": [[1, 1, 384, 384], [1, 1, 384, 384], [8, 1, 384, 384]]
}
else:
common_dynamic_shapes = None