> 文章列表 > onnx笔记1: onnx读写、创建以及修改

onnx笔记1: onnx读写、创建以及修改

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.cconnx_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