diff --git a/ppocr/utils/export_model.py b/ppocr/utils/export_model.py index ccb4456eca..3394c20c79 100644 --- a/ppocr/utils/export_model.py +++ b/ppocr/utils/export_model.py @@ -241,6 +241,14 @@ def dynamic_to_static(model, arch_config, logger, input_shape=None): ], full_graph=True, ) + elif arch_config["algorithm"] == "SLANeXt": + model = paddle.jit.to_static( + model, + input_spec=[ + paddle.static.InputSpec(shape=[-1, 1, 512, 512], dtype="float32") + ], + full_graph=True, + ) elif arch_config["algorithm"] == "PP-FormulaNet-L": model = paddle.jit.to_static( model,