首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >Tensorflow 2.0模型检查点文件到.pb/onnx文件

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

Stack Overflow用户
提问于 2021-08-17 00:10:26
回答 1查看 35关注 0票数 0

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

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2021-08-24 17:39:52

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

票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/68810364

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档