tensorflow转onnx的实现方法
更新时间：2023年03月06日 09:58:56 作者：小广向前进
本文主要介绍了tensorflow转onnx的实现方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
安装tf2onnx以及onnxruntime
pip install onnxruntime pip install tf2onnx
tf 转为onnx步骤为如下：
- 将tf动态图冻结，生成冻结后的pb文件
- 使用tf2onnx 将pb文件转为onnx文件
将tf动态图冻结使用如下代码：
def export_frozen_graph(model, model_dir, name_pb) :
f = tf.function(lambda x: model(inputs=x))
f = f.get_concrete_function(x=(tf.TensorSpec(model.inputs[0].shape, model.inputs[0].dtype)))
frozen_func = convert_variables_to_constants_v2(f)
frozen_func.graph.as_graph_def()
print("-" * 50)
print("Frozen model inputs: ")
print(frozen_func.inputs)
print("Frozen model outputs: ")
print(frozen_func.outputs)
tf.io.write_graph(graph_or_graph_def=frozen_func.graph,
logdir=model_dir,
name=name_pb,
as_text=False)使用tf2onnx 将pb文件转为onnx文件时需要在终端执行，需要指出的是大部分tf模型的输入layout都是NHWC，而ONNX模型的输入layout为NCHW，因此建议在转换的时候加上--inputs-as-nchw这个选项，其他选项可以参考文档，非常详细，具体运行命令如下：
python -m tf2onnx.convert --input yolo.pb --output model.onnx --outputs Identity:0,Identity_1:0,Identity_2:0 --inputs x:0 --inputs-as-nchw x:0 --opset 10
参数说明：
- input 输入的pb模型
- output 输出的onnx文件名
- inputs 输入层名字，有多个输入时，中间用逗号隔开
- outputs 输出层名字，有多个输出时，中间用逗号隔开
- –inputs-as-nchw 将输入作为nchw格式，注意加上输入层名字
- –opset onnx 版本号
通过程序直接转：
tf2onnx.convert.from_keras(model, inputs_as_nchw=[model.inputs[0].name], output_path=model_filepath + 'yolo.onnx') --opset 10
到此这篇关于tensorflow转onnx的实现方法的文章就介绍到这了,更多相关tensorflow转onnx内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
Python的3.0版本，常被称为Python 3000，或简称Py3k。这篇文章主要介绍了python 3.0 模拟用户登录功能并实现三次错误锁定,需要的朋友可以参考下2017-11-11
这篇文章主要为大家详细介绍了如何利用Python和Pygame实现24点小游戏，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2022-04-04
这篇文章主要介绍了Python使用描述符实现属性类型检查,实例属性就是在一个类中将另一个类的实例作为该类的一个数属性，本文通过代码演示给大家介绍的非常详细，需要的朋友可以参考下2022-05-05
今天小编就为大家分享一篇python获取交互式ssh shell的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-02-02
这篇文章主要介绍了Python使用xlwt模块操作Excel的方法,结合实例形式分析了Python安装xlwt模块及使用xlwt模块针对Excel文件的创建、设置、保存等常用操作技巧,需要的朋友可以参考下2018-03-03
这篇文章主要介绍了关于Numpy之repeat、tile的用法总结，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2021-06-06
你可以直接在 NumPy 数组之间使用算术运算符 + - * /,但本节讨论了一个扩展,其中我们有函数可以接受任何类似数组的对象,如列表、元组等,并根据条件执行算术运算,文中通过代码示例给大家讲解的非常详细,需要的朋友可以参考下2024-06-06
一个人可能会在计算机上存储大量的照片、视频和文档文件，这些文件可能散落在不同的文件夹中，难以管理和查找。所以本文就来用Python制作一个自动化文档整理工具吧2023-04-04
Python SQLAlchemy 是一个强大且多功能的 Python SQL 工具包和对象关系映射 (ORM) 系统,提供了一整套众所周知的企业级持久性模式,本文为大家整理了它必须了解的3个数据库操作功能,希望对大家有所帮助2023-09-09
这篇文章主要介绍了python 用pandas实现数据透视表功能的方法，帮助大家更好的理解和使用python，感兴趣的朋友可以了解下2020-12-12
最新评论