Support variable length input

2022-08-15 09:00:35 +00:00 · 2022-08-15 09:00:35 +00:00 · 7efee821bc
parent ebe3e885c0
commit 7efee821bc
1 changed files with 2 additions and 0 deletions
--- a/tools/export_model.py
+++ b/tools/export_model.py
@ -140,6 +140,8 @@ def export_single_model(model,
            infer_shape = [3, 488, 488]
            if arch_config["algorithm"] == "TableMaster":
                infer_shape = [3, 480, 480]
+            if arch_config["algorithm"] == "SLANet":
+                infer_shape = [3, -1, -1]
        model = to_static(
            model,
            input_spec=[