1.pip install tf2onnx
2.python -m tf2onnx.covert 会出现使用样例,根据需要选择使用
3.命令参数需要知道输入节点名字和输出节点的名字
4.出现unsupported onnx opset version:11错误,解决方法:在命令最后加上 --opset 11
例如:python -m tf2onnx.convert --checkpoint cats_dogs.ckpt.meta --inputs Placeholder:0 --outputs output:0 --output onnxModel.onnx --opset 11 (模型有多个输入多个输入的情况,之间用,隔开)
参考网页:https://github.com/onnx/tensorflow-onnx/issues/886
最后
以上就是完美火最近收集整理的关于tensorflow模型使用tf2onnx转onnx模型方法的全部内容,更多相关tensorflow模型使用tf2onnx转onnx模型方法内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复