使用tf2onnx在Python中将TensorFlow模型转换为ONNX

Question

是否可以安装tf2onnx，例如使用conda以编程方式在Python中转换TF模型。例如：

from tf2onnx import convert
convert(model_dir)

Answer 1

是的，这是可能的。 您也可以进行pip安装，要将模型转换为ONNX格式，请参考以下示例：https://github.com/onnx/tutorials/blob/master/tutorials/TensorflowToOnnx-1.ipynb