提交 cacf8f8a 编写于 作者: C Channingss

export_model support dynamic input shape

上级 d1e150e2
...@@ -51,8 +51,11 @@ def main(): ...@@ -51,8 +51,11 @@ def main():
model.eval() model.eval()
save_path = '{}/inference'.format(config['Global']['save_inference_dir']) 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 = to_static(
model, model,
input_spec=[ input_spec=[
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册