From 2b7b76310b2fb386a2fea477a4481073dd531f33 Mon Sep 17 00:00:00 2001 From: zhangyubo0722 <94225063+zhangyubo0722@users.noreply.github.com> Date: Tue, 25 Feb 2025 14:59:49 +0800 Subject: [PATCH] fix formula rec models hpi_config (#14739) 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. --- ppocr/utils/export_model.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/ppocr/utils/export_model.py b/ppocr/utils/export_model.py index ad49dd5913..725c4aa073 100644 --- a/ppocr/utils/export_model.py +++ b/ppocr/utils/export_model.py @@ -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