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

fix sdk export pipeline.json (#1391)

Browse Source
This commit is contained in:
Chen Xin 2022-11-18 16:46:58 +08:00 committed by GitHub
parent 115ca470dd
commit a59b17259d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -245,8 +245,9 @@ def get_pipeline(deploy_cfg: mmengine.Config, model_cfg: mmengine.Config,
task = get_task_type(deploy_cfg) task = get_task_type(deploy_cfg)
input_names = preprocess['input'] input_names = preprocess['input']
output_names = postprocess['output'] output_names = postprocess['output']
if task == Task.CLASSIFICATION or task == Task.SUPER_RESOLUTION \ if task in [
or Task.VIDEO_RECOGNITION: Task.CLASSIFICATION, Task.SUPER_RESOLUTION, Task.VIDEO_RECOGNITION
]:
postprocess['input'] = infer_info['output'] postprocess['input'] = infer_info['output']
else: else:
postprocess['input'] = preprocess['output'] + infer_info['output'] postprocess['input'] = preprocess['output'] + infer_info['output']