From 7efee821bcd5feda5ca1aade47ed954c3252cf76 Mon Sep 17 00:00:00 2001
From: WenmuZhou <572459439@qq.com>
Date: Mon, 15 Aug 2022 09:00:35 +0000
Subject: [PATCH] Support variable length input

---
 tools/export_model.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tools/export_model.py b/tools/export_model.py
index 2443d66c..3d30fa77 100755
--- 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=[
-- 
GitLab