tensorflow - 如何将 TensorFlow v2 模型转换为 Onnx

Question

我正在尝试将我的 TF 2.0 RC1 模型转换为 onnx（使用 keras2onnx 或 tf2onnx）。当我执行此代码时：

import keras2onnx

或此代码

import tf2onnx

我得到错误：

ModuleNotFoundError: No module named 'tensorflow.tools.graph_transforms'

如何解决此错误或以其他方式将我的 Tensorflow v2.0 模型转换为 onnx？

Answer 1

这应该过去了。

pip install git+https://github.com/onnx/tensorflow-onnx