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文章来源:https://www.toymoban.com/news/detail-421640.html
打开onnx-ml.proto
代码,可以看到NodeProto
的定义文章来源地址https://www.toymoban.com/news/detail-421640.html
message NodeProto {
repeated string input = 1; // namespace Valu
到了这里,关于onnx笔记1: onnx读写、创建以及修改的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!