[Enhancement] Additional arguments support for OpenVINO Model Optimiz…

…er (open-mmlab#178) * Add mo args. * [Docs]: update docs and argument descriptions (open-mmlab#196) * bump version to v0.4.0 * update docs and argument descriptions * revert version change * fix unnecessary change of config for dynamic exportation (open-mmlab#199) * fix mmcls get classes (open-mmlab#215) * fix mmcls get classes * resolve comment * resolve comment * Add ModelOptimizerOptions. * Fix merge bugs. * Update mmpose.md (open-mmlab#224) * [Dostring]add example in apis docstring (open-mmlab#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 <[email protected]> Co-authored-by: Haofan Wang <[email protected]> Co-authored-by: VVsssssk <[email protected]> Co-authored-by: grimoire <[email protected]>
irexyc · Mar 30, 2022 · 9462e27 · 9462e27
1 parent 7b9efc8
commit 9462e27
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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'