Python onnx 模型打印显示所有节点及查看相互关系

最新推荐文章于 2024-12-21 17:23:34 发布

AI算法网奇

最新推荐文章于 2024-12-21 17:23:34 发布

阅读量834

点赞数

CC 4.0 BY-SA版权

分类专栏： python基础文章标签：数学建模

原文链接：https://blog.youkuaiyun.com/tanmx219/article/details/120898052

python基础专栏收录该内容

863 篇文章

订阅专栏

文章介绍了在处理ONNX模型时，如何通过PythonAPI获取并打印出所有节点的信息和权重参数。作者分享了简单的代码示例，包括遍历节点输出和初始化器（权重）的详细信息，特别查找了特定输出名称的节点。此外，还展示了获取模型所有输出节点的方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最近使用onnx时，想把所有的节点的信息和权重参数显示出来，找了一下没找到类似的源码，官方介绍的pythonAPI都是些什么加载，保存，转换之类的，没有详细介绍怎么使用onnx分析模型的，只好自己写一个。

其实很简单，我只列些最基本的，具体分析还得看个人的需要，

import onnx
if __name__ == '__main__':

    model_path = r'F:\model_float32.onnx'

    model = onnx.load(model_path)

    nodes = model.graph.node
    nodnum = len(nodes)  # 205

    for nid in range(nodnum):
        if (nodes[nid].output[0] == 'stride_32'):
            print('Found stride_32: index = ', nid)
        else:
            print(nodes[nid].output)

    inits = model.graph.initializer
    ininum = len(inits)  # 124

    for iid in range(ininum):
        el = inits[iid]
        print('name:', el.name, ' dtype:', el.data_type, ' dim:', el.dims)  # el.raw_data for weights and biases

    print(model.graph.output)  # display all the output nodes