fix SLANeXt export bug (#14512)

* add slanext models * refine codes * refine codes * refine codes * fix export SLANeXt
PaddlePaddle · Jan 7, 2025 · a6b96bb · a6b96bb
1 parent 359ab6c
commit a6b96bb
Showing 1 changed file with 8 additions and 0 deletions.
diff --git 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,