add is_batched argument to pipeline.json (#1528)

open-mmlab · Mar 1, 2023 · bda0960 · bda0960
1 parent 11ca21a
commit bda0960
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mmdeploy/backend/sdk/export_info.py b/mmdeploy/backend/sdk/export_info.py
@@ -148,10 +148,12 @@ def get_inference_info(deploy_cfg: mmengine.Config, model_cfg: mmengine.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='Task',
         module='Net',
+        is_batched=is_batched,
         input=['prep_output'],
         output=['infer_output'],
         input_map=input_map,