From 91129c816fa42eeff4e34aaedc696d9db2bc6381 Mon Sep 17 00:00:00 2001 From: Semyon Bevzyuk Date: Tue, 15 Mar 2022 05:18:58 +0300 Subject: [PATCH] [Enhancement] Additional arguments support for OpenVINO Model Optimizer (#178) * Add mo args. * [Docs]: update docs and argument descriptions (#196) * bump version to v0.4.0 * update docs and argument descriptions * revert version change * fix unnecessary change of config for dynamic exportation (#199) * fix mmcls get classes (#215) * fix mmcls get classes * resolve comment * resolve comment * Add ModelOptimizerOptions. * Fix merge bugs. * Update mmpose.md (#224) * [Dostring]add example in apis docstring (#214) * add example in apis docstring * add backend example in docstring * rm blank line * Fixed get_mo_options_from_cfg args * fix l2norm test Co-authored-by: RunningLeon Co-authored-by: Haofan Wang Co-authored-by: VVsssssk <88368822+VVsssssk@users.noreply.github.com> Co-authored-by: grimoire --- mmdeploy/utils/test.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mmdeploy/utils/test.py b/mmdeploy/utils/test.py index 71ff7eec59..6f1decaa6b 100644 --- a/mmdeploy/utils/test.py +++ b/mmdeploy/utils/test.py @@ -517,8 +517,9 @@ def get_backend_outputs(ir_file_path: str, name: value.shape for name, value in flatten_model_inputs.items() } + mo_options = get_mo_options_from_cfg(deploy_cfg) openvino_apis.onnx2openvino(input_info, output_names, ir_file_path, - openvino_work_dir) + openvino_work_dir, mo_options) backend_files = [openvino_file_path] backend_feats = flatten_model_inputs device = 'cpu'