mirrors/mmdeploy
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
Code Issues Projects Releases Wiki Activity

add is_batched argument to pipeline.json (#1528)

Browse Source
This commit is contained in:
AllentDan 2022-12-20 17:37:47 +08:00 committed by GitHub
parent 84ee45bc35
commit acae7a756e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmdeploy/backend/sdk/export_info.py
View File

@ -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_name = input_names[0] if input_names else 'input'
input_map = dict(img=input_name) input_map = dict(img=input_name)
output_map = {} output_map = {}
is_batched = is_dynamic_batch(deploy_cfg, input_name=input_map['img'])
return_dict = dict( return_dict = dict(
name=name, name=name,
type='Task', type='Task',
module='Net', module='Net',
is_batched=is_batched,
input=['prep_output'], input=['prep_output'],
output=['infer_output'], output=['infer_output'],
input_map=input_map, input_map=input_map,