onnx笔记1: onnx读写、创建以及修改
1. 介绍
1.1 onnx proto
onnx存储pytorch或者tensorflow等框架的模型结构,本文了解onnx是如何存储这些模型结构以及onnx文件的形式。
- onnx本质是一种
protobuf
格式文件 - onnx通过
onnx-ml.proto
来描述onnx的结构,利用protoc编译程序,将onnx结构编译成onnx-ml.pb.cc
和onnx_ml_pb2.py
文件。 - 然后利用
onnx-ml.pb.cc
和代码操作onnx模型文件,实现对onnx模型文件进行增删改
onnx-ml.proto
的github地址: https://github.com/onnx/onnx/blob/main/onnx/onnx-ml.proto
打开onnx-ml.proto
代码,可以看到NodeProto
的定义
message NodeProto {repeated string input = 1; // namespace Valu