openvino2tensorflow 项目使用教程

openvino2tensorflow 项目使用教程

openvino2tensorflowThis script converts the ONNX/OpenVINO IR model to Tensorflow's saved_model, tflite, h5, tfjs, tftrt(TensorRT), CoreML, EdgeTPU, ONNX and pb. PyTorch (NCHW) -> ONNX (NCHW) -> OpenVINO (NCHW) -> openvino2tensorflow -> Tensorflow/Keras (NHWC/NCHW) -> TFLite (NHWC/NCHW). And the conversion from .pb to saved_model and from saved_model to .pb and from .pb to .tflite and saved_model to .tflite and saved_model to onnx. Support for building environments with Docker. It is possible to directly access the host PC GUI and the camera to verify the operation. NVIDIA GPU (dGPU) support. Intel iHD GPU (iGPU) support.项目地址:https://gitcode.com/gh_mirrors/op/openvino2tensorflow

1. 项目的目录结构及介绍

openvino2tensorflow/
├── README.md
├── requirements.txt
├── setup.py
├── openvino2tensorflow/
│   ├── __init__.py
│   ├── converter.py
│   ├── utils.py
│   └── config.py
└── tests/
    ├── test_converter.py
    └── test_utils.py

目录结构介绍

• README.md: 项目的基本介绍和使用说明。
• requirements.txt: 项目依赖的Python包列表。
• setup.py: 用于安装项目的脚本。
• openvino2tensorflow/: 项目的主要代码目录。
  • init.py: 初始化文件，使该目录成为一个Python包。
  • converter.py: 核心转换脚本，负责将OpenVINO IR模型转换为TensorFlow模型。
  • utils.py: 工具函数脚本，包含一些辅助函数。
  • config.py: 配置文件脚本，包含项目的配置参数。
• tests/: 测试代码目录，包含项目的单元测试。
  • test_converter.py: 针对converter.py的单元测试。
  • test_utils.py: 针对utils.py的单元测试。

2. 项目的启动文件介绍

项目的启动文件是openvino2tensorflow/converter.py。该文件包含了主要的转换逻辑，负责将OpenVINO IR模型转换为TensorFlow的saved_model、tflite、h5和pb格式。

主要功能

• 模型转换: 将OpenVINO IR模型转换为TensorFlow模型。
• 格式支持: 支持多种输出格式，包括saved_model、tflite、h5和pb。
• 配置加载: 从config.py中加载配置参数，用于控制转换过程。

使用方法

from openvino2tensorflow import converter

# 加载OpenVINO IR模型
ir_model_path = "path/to/your/model.xml"

# 转换模型
converter.convert(ir_model_path)

3. 项目的配置文件介绍

项目的配置文件是openvino2tensorflow/config.py。该文件包含了项目的配置参数，用于控制转换过程中的各种选项。

配置参数

• output_format: 指定输出模型的格式，可选值包括saved_model、tflite、h5和pb。
• output_dir: 指定输出模型的保存路径。
• input_shape: 指定输入模型的形状，格式为(N, C, H, W)。
• batch_size: 指定批处理大小。

示例配置

# config.py

output_format = "saved_model"
output_dir = "output_models"
input_shape = (1, 3, 224, 224)
batch_size = 1

使用方法

在启动文件converter.py中，可以通过以下方式加载配置：

from openvino2tensorflow import config

# 加载配置
output_format = config.output_format
output_dir = config.output_dir
input_shape = config.input_shape
batch_size = config.batch_size

通过以上配置，可以灵活地控制模型的转换过程。

openvino2tensorflowThis script converts the ONNX/OpenVINO IR model to Tensorflow's saved_model, tflite, h5, tfjs, tftrt(TensorRT), CoreML, EdgeTPU, ONNX and pb. PyTorch (NCHW) -> ONNX (NCHW) -> OpenVINO (NCHW) -> openvino2tensorflow -> Tensorflow/Keras (NHWC/NCHW) -> TFLite (NHWC/NCHW). And the conversion from .pb to saved_model and from saved_model to .pb and from .pb to .tflite and saved_model to .tflite and saved_model to onnx. Support for building environments with Docker. It is possible to directly access the host PC GUI and the camera to verify the operation. NVIDIA GPU (dGPU) support. Intel iHD GPU (iGPU) support.项目地址:https://gitcode.com/gh_mirrors/op/openvino2tensorflow