Merge pull request #1837 from Channingss/dygraph

export_model support dynamic input shape
4 years ago · 4b715cf200
parent 4d8f7e6ce2 cacf8f8a6c
commit 4b715cf200
1 changed files with 5 additions and 2 deletions
--- 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=[