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