Tensorflow 2.0 模型检查点文件到 .pb/onnx 文件
Tensorflow 2.0 model checkpoint files to .pb/onnx file
对于TF-2.0生成的模型检查点文件(通常由.meta
、.data
、.index
组成),如何将其转换为onnx
或pb
文件?
因为我发现现有的大部分工具,比如tf2onnx
只支持TF-1.x.
tf2onnx 支持 tf2、保存的模型和检查点文件。我建议制作一个保存的模型:
model.save("path")
然后使用 tf2onnx 命令行:
tf2onnx.convert --saved-model path/to/model --output model.onnx --opset 12
对于TF-2.0生成的模型检查点文件(通常由.meta
、.data
、.index
组成),如何将其转换为onnx
或pb
文件?
因为我发现现有的大部分工具,比如tf2onnx
只支持TF-1.x.
tf2onnx 支持 tf2、保存的模型和检查点文件。我建议制作一个保存的模型:
model.save("path")
然后使用 tf2onnx 命令行:
tf2onnx.convert --saved-model path/to/model --output model.onnx --opset 12