add 'is_batched' argument to pipeline.json (#1560)

2022-12-21 10:45:00 +08:00 · 2022-12-21 10:45:00 +08:00 · 26d71ce0a8
parent f9f351ae1f
commit 26d71ce0a8
1 changed files with 2 additions and 0 deletions
--- a/mmdeploy/backend/sdk/export_info.py
+++ b/mmdeploy/backend/sdk/export_info.py
@ -220,10 +220,12 @@ def get_inference_info(deploy_cfg: mmcv.Config, model_cfg: mmcv.Config,
        input_name = input_names[0] if input_names else 'input'
        input_map = dict(img=input_name)
        output_map = {}
+    is_batched = is_dynamic_batch(deploy_cfg, input_name=input_map['img'])
    return_dict = dict(
        name=name,
        type=type,
        module=module,
+        is_batched=is_batched,
        input=input,
        output=output,
        input_map=input_map,