From 1a1ebd7542467b0ffa284357bb1305613540d80b Mon Sep 17 00:00:00 2001 From: Semyon Bevzyuk Date: Mon, 31 Jan 2022 23:54:40 +0300 Subject: [PATCH] Fix capture_output --- docs/en/tutorials/how_to_support_new_backends.md | 2 +- mmdeploy/backend/openvino/onnx2openvino.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/en/tutorials/how_to_support_new_backends.md b/docs/en/tutorials/how_to_support_new_backends.md index 738b02a9ae..85319b8d91 100644 --- a/docs/en/tutorials/how_to_support_new_backends.md +++ b/docs/en/tutorials/how_to_support_new_backends.md @@ -94,7 +94,7 @@ The backends in MMDeploy must support the ONNX. The backend loads the ".onnx" fi f'--input_shape="{input_shapes}" ' \ f'--disable_fusing ' command = f'mo.py {mo_args}' - mo_output = run(command, capture_output=True, shell=True, check=True) + mo_output = run(command, stdout=PIPE, stderr=PIPE, shell=True, check=True) ``` **Use executable program:** diff --git a/mmdeploy/backend/openvino/onnx2openvino.py b/mmdeploy/backend/openvino/onnx2openvino.py index 201ef76b2f..eb592cc1c0 100644 --- a/mmdeploy/backend/openvino/onnx2openvino.py +++ b/mmdeploy/backend/openvino/onnx2openvino.py @@ -1,7 +1,7 @@ # Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import subprocess -from subprocess import CalledProcessError, run +from subprocess import PIPE, CalledProcessError, run from typing import Dict, List, Union import mmcv @@ -86,7 +86,7 @@ def onnx2openvino(input_info: Dict[str, Union[List[int], torch.Size]], logger = get_root_logger() logger.info(f'Args for Model Optimizer: {command}') - mo_output = run(command, capture_output=True, shell=True, check=True) + mo_output = run(command, stdout=PIPE, stderr=PIPE, shell=True, check=True) logger.info(mo_output.stdout.decode()) logger.debug(mo_output.stderr.decode())