openvino使用过程中 IR 文件转换遇到 Error

部署运行你感兴趣的模型镜像

 

在使用 mo_tf.py 文件对 Tensorflow 模型进行转换时:

python mo_tf.py --input_model E:\WORK\vino\tensorflow\ssd_mobilenet_v1_coco_2018_01_28\frozen_inference_graph.pb --tensorflow_use_custom_operations_config C:\Intel\openvino\deployment_tools\model_optimizer\extensions\front\tf\ssd_support.json

总是报以下错误

官网的解释是:

实测解决方案:

pip install networkx==2.3

再次运行转换脚本即可成功运行

您可能感兴趣的与本文相关的镜像

TensorFlow-v2.15

TensorFlow-v2.15

TensorFlow

TensorFlow 是由Google Brain 团队开发的开源机器学习框架,广泛应用于深度学习研究和生产环境。 它提供了一个灵活的平台,用于构建和训练各种机器学习模型

### OpenVINO 2022 中 ONNX 到 IR 的转换与最佳实践 OpenVINO 2022 版本提供了强大的工具链来支持 ONNX 模型的导入和优化。通过 `mo.py`(Model Optimizer)脚本,可以轻松实现从 ONNX 格式到 OpenVINO Intermediate Representation (IR) [.xml 和 .bin] 的转换[^1]。 #### 转换流程概述 为了将 ONNX 模型成功转换为 OpenVINO IR 格式,需遵循以下技术细节: - **安装依赖项**:确保已正确配置 OpenVINO 环境并完成必要的 Python 库安装。例如,在 Windows 平台上可以通过官方文档指导完成安装过程[^3]。 - **调用 Model Optimizer**:使用命令行接口运行 `mo.py` 脚本来执行转换操作。以下是典型的命令结构: ```bash mo --input_model <path_to_onnx_model> --output_dir <output_directory> ``` 此命令会读取指定路径下的 ONNX 文件,并将其转化为 `.xml` 和 `.bin` 文件形式存储于目标目录下[^2]。 - **参数调整**:依据具体应用场景可能需要额外设置一些选项以满足特定需求,比如输入形状定义 (`--input_shape`) 或数据类型精度控制 (`--data_type FP16|FP32`) 等。 #### 示例代码展示 下面给出一段完整的Python脚本用于自动化上述转换步骤: ```python import os from subprocess import run, PIPE def convert_onnx_to_ir(onnx_path, ir_output_dir): command = [ 'mo', '--input_model', onnx_path, '--output_dir', ir_output_dir ] result = run(command, stdout=PIPE, stderr=PIPE) if result.returncode != 0: raise Exception(f"Conversion failed with error {result.stderr.decode()}") if __name__ == "__main__": current_dir = os.getcwd() model_file = os.path.join(current_dir, "model.onnx") output_folder = os.path.join(current_dir, "ir_models") try: convert_onnx_to_ir(model_file, output_folder) print("ONNX to IR conversion completed successfully.") except Exception as e: print(e) ``` 以上脚本封装了一个函数 `convert_onnx_to_ir()` 来简化批量处理多个模型文件的过程。 #### 性能考量与建议 当考虑部署效率时,推荐采用较低精度的数据表示方式如 INT8 或者混合量化策略,这样可以在保持较高准确性的同时显著减少计算资源消耗以及加速推理速度。
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值