[深度学习主流框架解析一] Onnx

1、模型文件协议解析

Onnx同样也是采用了Protobuf协议进行模型文件的整体构建，与Caffe相比，Onnx使用了多proto文件描述模型。因此，解析Onnx模型文件需要了解多个proto文件之间的关联。

总共的proto文件如下所示：

onnx-data.proto
onnx-ml.proto
onnx-operators.proto
onnx-operators-ml.proto
onnx.proto

proto文件之间的依赖关系如下所示：

#mermaid-svg-lo1rMhf2hypVwSNL {font-family:"trebuchet ms",verdana,arial,sans-serif;font-size:16px;fill:#333;}#mermaid-svg-lo1rMhf2hypVwSNL .error-icon{fill:#552222;}#mermaid-svg-lo1rMhf2hypVwSNL .error-text{fill:#552222;stroke:#552222;}#mermaid-svg-lo1rMhf2hypVwSNL .edge-thickness-normal{stroke-width:2px;}#mermaid-svg-lo1rMhf2hypVwSNL .edge-thickness-thick{stroke-width:3.5px;}#mermaid-svg-lo1rMhf2hypVwSNL .edge-pattern-solid{stroke-dasharray:0;}#mermaid-svg-lo1rMhf2hypVwSNL .edge-pattern-dashed{stroke-dasharray:3;}#mermaid-svg-lo1rMhf2hypVwSNL .edge-pattern-dotted{stroke-dasharray:2;}#mermaid-svg-lo1rMhf2hypVwSNL .marker{fill:#333333;stroke:#333333;}#mermaid-svg-lo1rMhf2hypVwSNL .marker.cross{stroke:#333333;}#mermaid-svg-lo1rMhf2hypVwSNL svg{font-family:"trebuchet ms",verdana,arial,sans-serif;font-size:16px;}#mermaid-svg-lo1rMhf2hypVwSNL .label{font-family:"trebuchet ms",verdana,arial,sans-serif;color:#333;}#mermaid-svg-lo1rMhf2hypVwSNL .cluster-label text{fill:#333;}#mermaid-svg-lo1rMhf2hypVwSNL .cluster-label span{color:#333;}#mermaid-svg-lo1rMhf2hypVwSNL .label text,#mermaid-svg-lo1rMhf2hypVwSNL span{fill:#333;color:#333;}#mermaid-svg-lo1rMhf2hypVwSNL .node rect,#mermaid-svg-lo1rMhf2hypVwSNL .node circle,#mermaid-svg-lo1rMhf2hypVwSNL .node ellipse,#mermaid-svg-lo1rMhf2hypVwSNL .node polygon,#mermaid-svg-lo1rMhf2hypVwSNL .node path{fill:#ECECFF;stroke:#9370DB;stroke-width:1px;}#mermaid-svg-lo1rMhf2hypVwSNL .node .label{text-align:center;}#mermaid-svg-lo1rMhf2hypVwSNL .node.clickable{cursor:pointer;}#mermaid-svg-lo1rMhf2hypVwSNL .arrowheadPath{fill:#333333;}#mermaid-svg-lo1rMhf2hypVwSNL .edgePath .path{stroke:#333333;stroke-width:2.0px;}#mermaid-svg-lo1rMhf2hypVwSNL .flowchart-link{stroke:#333333;fill:none;}#mermaid-svg-lo1rMhf2hypVwSNL .edgeLabel{background-color:#e8e8e8;text-align:center;}#mermaid-svg-lo1rMhf2hypVwSNL .edgeLabel rect{opacity:0.5;background-color:#e8e8e8;fill:#e8e8e8;}#mermaid-svg-lo1rMhf2hypVwSNL .cluster rect{fill:#ffffde;stroke:#aaaa33;stroke-width:1px;}#mermaid-svg-lo1rMhf2hypVwSNL .cluster text{fill:#333;}#mermaid-svg-lo1rMhf2hypVwSNL .cluster span{color:#333;}#mermaid-svg-lo1rMhf2hypVwSNL div.mermaidTooltip{position:absolute;text-align:center;max-width:200px;padding:2px;font-family:"trebuchet ms",verdana,arial,sans-serif;font-size:12px;background:hsl(80, 100%, 96.2745098039%);border:1px solid #aaaa33;border-radius:2px;pointer-events:none;z-index:100;}#mermaid-svg-lo1rMhf2hypVwSNL :root{--mermaid-font-family:"trebuchet ms",verdana,arial,sans-serif;}

onnx.proto

onnx-operators.proto

onnx-ml.proto

onnx-data.proto

onnx-operators-ml.proto

1.1、onnx.proto说明

整体的模型文件可见下图。

1.2、onnx-operators.proto说明

主要定义了OperatorProto和OperatorSetProto这两种消息类型，用于传递特定版本的和具体类型的网络层算子参数

1.3、onnx-ml.proto和onnx-operators-ml.proto说明

onnx-ml.proto等同于onnx.proto
onnx-operators-ml.proto等同于onnx-operators.proto
内部内容基本一致

1.4、onnx-data.proto说明

增加了SequenceProto、MapProto等基础的数据类型
增加了OptionalProto去包含所有的特殊的数据类型