如何将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