export_model support dynamic input shape

cacf8f8a · Channingss · d1e150e2 · cacf8f8a
Commit cacf8f8a authored Jan 26, 2021 by Channingss
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 2 deletions

tools/export_model.py tools/export_model.py +5 -2

No files found.
--- a/tools/export_model.py
+++ b/tools/export_model.py
@@ -51,8 +51,11 @@ def main():
    model.eval()

    save_path = '{}/inference'.format(config['Global']['save_inference_dir'])
-    infer_shape = [3, 32, 100] if config['Architecture'][
-        'model_type'] != "det" else [3, 640, 640]
+
+    infer_shape = [3, -1, -1] 
+    if config['Architecture']['model_type'] == "rec":
+         infer_shape = [3, 32, -1]
+
    model = to_static(
        model,
        input_spec=[