Tensorflow 2.0 模型检查点文件到 .pb/onnx 文件

Tensorflow 2.0 model checkpoint files to .pb/onnx file

对于TF-2.0生成的模型检查点文件(通常由.meta.data.index组成),如何将其转换为onnxpb文件? 因为我发现现有的大部分工具,比如tf2onnx只支持TF-1.x.

tf2onnx 支持 tf2、保存的模型和检查点文件。我建议制作一个保存的模型:

model.save("path")

然后使用 tf2onnx 命令行:

tf2onnx.convert --saved-model path/to/model --output model.onnx --opset 12

https://github.com/onnx/tensorflow-onnx#--saved-model