diff --git a/ppocr/utils/export_model.py b/ppocr/utils/export_model.py index 9d2f76639..88f6e50ea 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, 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 = {