From 17fff8cca47201617a457219df129e7a17a12543 Mon Sep 17 00:00:00 2001 From: Tingquan Gao Date: Tue, 11 Feb 2025 11:17:23 +0800 Subject: [PATCH] fix dy shapes of trt for rec models (#14654) --- ppocr/utils/export_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppocr/utils/export_model.py b/ppocr/utils/export_model.py index 51ab82e5bd..ad49dd5913 100644 --- a/ppocr/utils/export_model.py +++ b/ppocr/utils/export_model.py @@ -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 = {