tensorRT--- 认识onnx

本文介绍了ONNX的本质——一种Protobuf格式文件,详细阐述了onnx-ml.proto在定义模型结构中的作用,并探讨了如何使用ONNX进行模型创建、读取和修改。重点内容包括PyTorch模型转ONNX、ONNX模型结构解析以及模型的操作步骤。
摘要由CSDN通过智能技术生成

1ONNX的本质,是一种Protobuf格式文件

2Protobuf则通过onnx-ml.proto编译得到onnx-ml.pb.honnx-ml.pb.cconnx_ml_pb2.py

3、然后用onnx-ml.pb.cc和代码来操作onnx模型文件,实现增删改

4onnx-ml.proto则是描述onnx文件如何组成的,具有什么结构,他是操作onnx经常参照的东西

 https://github.com/onnx/onnx/blob/main/onnx/onnx-ml.proto

表示onnx中有节点类型叫node

他有input属性,是repeated,即重复类型,数组

他有output属性,是repeated,即重复类型,数组

他有name属性是string类型

对于repeated是数组,对于optional无视他

对于input = 1,后面的数字是id,无视他

我们只关心是否数组,类型是什么

model表示整个onnx的模型,包含图结构和解析器格式、opset版本、导出程序类型model.graph表示图结构,通常是我们netron看到的主要结构model.graph.node

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值