tensorflow转onnx如何实现
导读：本文共1148字符，通常情况下阅读需要4分钟。同时您也可以点击右侧朗读，来听本文内容。按键盘←（左） →（右） 方向键可以翻页。
摘要： 安装tf2onnx以及onnxruntimepipinstallonnxruntimepipinstalltf2onnxtf 转为onnx步骤为如下：将tf动态图冻结，生成冻结后的pb文件使用tf2onnx 将pb文件转为onnx文件将tf动态图冻结使用如下代码：defexport_frozen_graph(model,model_dir,name_pb):... ...
音频解说
目录
(为您整理了一些要点),点击可以直达。
安装tf2onnx以及onnxruntime
tf 转为onnx步骤为如下：
将tf动态图冻结，生成冻结后的pb文件
使用tf2onnx 将pb文件转为onnx文件
将tf动态图冻结使用如下代码：
使用tf2onnx 将pb文件转为onnx文件时需要在终端执行，需要指出的是大部分tf模型的输入layout都是NHWC，而ONNX模型的输入layout为NCHW，因此建议在转换的时候加上--inputs-as-nchw这个选项，其他选项可以参考文档，非常详细，具体运行命令如下：
参数说明：
input 输入的pb模型
output 输出的onnx文件名
inputs 输入层名字，有多个输入时，中间用逗号隔开
outputs 输出层名字，有多个输出时，中间用逗号隔开
–inputs-as-nchw 将输入作为nchw格式，注意加上输入层名字
–opset onnx 版本号
通过程序直接转：
本文：
tensorflow转onnx如何实现的详细内容，希望对您有所帮助，信息来源于网络。