diff --git a/tools/export_model.py b/tools/export_model.py index 2443d66ca2a5d81cdb99964bfb88af29ae0c66e2..3d30fa77eac107f38913aa04dcf6c32f59f176ab 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=[