主要内容

exportONNXNetwork

导出网络到ONNX模型格式

描述

例子

exportONNXNetwork (文件名导出深度学习网络与ONNX™格式文件的权重文件名.如果文件名存在,那么exportONNXNetwork覆盖文件。

此函数需要ONNX模型格式的深度学习工具箱™转换器金宝app支持包。如果没有安装此支金宝app持包,则该函数将提供下载链接。

exportONNXNetwork (文件名名称=值使用由一个或多个名称-值参数指定的附加选项导出网络。

例子

全部折叠

加载预训练的SqueezeNet卷积神经网络。

网=挤压网
net = DAGNetwork with properties: Layers: [68×1 nnet.cnn.layer.Layer] Connections: [75×2 table] InputNames: {'data'} OutputNames: {'ClassificationLayer_predictions'}

导出网络作为一个ONNX格式的文件称为squeezenet.onnx.将文件保存到当前文件夹。如果没有安装ONNX模型格式的深度学习工具箱转换器支持包,则金宝appexportONNXNetwork在外接程序资源管理器中提供到所需支持包的链接。金宝app要安装支持包,请单击链接,然后金宝app单击安装

文件名=“squeezenet.onnx”;exportONNXNetwork(净、文件名)

现在可以导入squeezenet.onnx文件导入到任何支持ONNX导入的深度学习框架。金宝app

将有或没有输出层的层图导出为ONNX格式exportONNXNetwork

加载一个预训练的SqueezeNet卷积神经网络,并将预训练的网络转换为层图。

网=挤压网;lgraph1 = layerGraph(net)
lgraph1 = LayerGraph与属性:图层:[68×1 nnet.cnn.layer.Layer]连接:[75×2 table] InputNames: {'data'} OutputNames: {'ClassificationLayer_predictions'}

分析层图。analyzeNetwork显示网络体系结构的交互式图形和包含有关网络层信息的表。您还可以在层图中检测错误和问题lgraph1然后导出为ONNX格式。lgraph1是无错误的。

analyzeNetwork (lgraph1)

导出图层图lgraph1作为一个ONNX格式的文件在当前文件夹称为squeezeLayers1.onnx

exportONNXNetwork (lgraph1“squeezeLayers1.onnx”

现在,您可以导入squeezeLayers1.onnx文件导入到任何支持ONNX导入的深度学习框架。金宝app

的输出层lgraph1

lgraph2 = removeLayers(lgraph1,lgraph1. layers (end).Name)
lgraph2 = LayerGraph with properties: Layers: [67×1 nnet.cnn.layer.Layer] Connections: [74×2 table] InputNames: {'data'} OutputNames: {1×0 cell}

层图分析lgraph2通过使用analyzeNetwork.层图分析检测丢失的输出层和未连接的输出。你仍然可以导出lgraph2转换为ONNX格式。

analyzeNetwork (lgraph2)

导出图层图lgraph2作为一个ONNX格式的文件在当前文件夹称为squeezeLayers2.onnx

exportONNXNetwork (lgraph2“squeezeLayers2.onnx”

现在,您可以导入squeezeLayers2.onnx文件导入到任何支持ONNX导入的深度学习框架。金宝app

输入参数

全部折叠

训练过的网络或网络层的图,指定为aSeriesNetworkDAGNetworkdlnetwork,或LayerGraph对象。

你可以得到一个训练有素的网络(SeriesNetworkDAGNetwork,或dlnetwork)在这些方面:

  • 导入一个预先训练好的网络。例如,使用googlenet函数。

  • 培养自己的人脉。使用trainNetwork训练一个SeriesNetworkDAGNetwork.使用自定义训练循环来训练dlnetwork

一个LayerGraph对象是网络层的图。该图的一些层参数可能是空的(例如,卷积层的权重和偏置,批归一化层的均值和方差)。在使用层图作为输入参数之前exportONNXNetwork,通过分配随机值初始化空参数。或者,您可以在导出之前执行以下操作之一:

  • 把一个LayerGraph对象的dlnetwork对象使用层图作为输入参数dlnetwork.空参数会自动初始化。

  • 把一个LayerGraph对象的DAGNetwork对象,使用trainNetwork.使用图层图作为输入参数trainNetwork

在导出到ONNX网络之前,您可以在训练网络或网络层图中检测错误和问题analyzeNetworkexportONNXNetwork需要SeriesNetworkDAGNetwork,dlnetwork对象是无错误的。exportONNXNetwork允许导出LayerGraph对象,其输出层缺失或未连接。

文件名称,指定为字符向量或字符串标量。

例子:“network.onnx”

名称-值参数

指定可选参数对为Name1 = Value1,…,以=家,在那里的名字参数名称和价值对应的值。名称-值参数必须出现在其他参数之后,但对的顺序无关紧要。

例子:exportONNXNetwork(网、文件名、NetworkName =“my_net”)导出网络并指定“my_net”作为保存的ONNX网络中的网络名。

存储在保存文件中的ONNX网络的名称,指定为字符向量或字符串标量。

例子:NetworkName = " my_squeezenet "

导出模型中要使用的ONNX操作符的版本,指定为范围内的正整数13 [6].如果默认操作符集不支持要导出的网络,请尝试使用较低版本。金宝app如果您将导出的网络导入到另一个框架,并且在导出过程中使用了导入器不支持的操作符集,则导入可能会失败。金宝app

要确保使用了适当的操作符集版本,请参考ONNX操作符文档[3].例如,OpsetVersion = 9出口的maxUnpooling2dLayerMaxUnpool-9ONNX算子。

例子:OpsetVersion = 6

ONNX网络的批量大小,指定为[]或者作为正整数。如果你指定BatchSize作为[], ONNX网络具有动态批处理大小。如果你指定BatchSize作为正整数k时,ONNX网络有固定的批大小为k

例子:BatchSize = 10

限制

  • exportONNXNetwork金宝app支持以下ONNX版本:

    • 该函数支持ONNX中间表示金宝app版本7。

    • 该函数支持ONNX操作符集金宝app6 ~ 14。

  • exportONNXNetwork不导出与网络训练相关的设置或属性,如训练选项、学习率因素或正则化因素。

  • 如果您导出的网络包含ONNX格式不支持的层(请参阅金宝appONNX导出支金宝app持的层),然后exportONNXNetwork保存一个占位符ONNX操作符来代替不支持的层,并返回一个警告。金宝app不能将带有占位符操作符的ONNX网络导入到其他深度学习框架中。

  • 因为MATLAB之间的架构差异®和ONNX,导出的网络可能与原始网络有不同的结构。

请注意

如果导入导出的网络,重新导入的网络的层数可能与原始网络不同,可能不受支持。金宝app

更多关于

全部折叠

支持的层数金宝appONNX出口

exportONNXNetwork可以导出以下内容:

提示

  • 您可以将经过训练的MATLAB深度学习网络(包括多个输入和多个输出)导出为ONNX模型格式。要了解多输入多输出深度学习网络,请参见多输入多输出网络

参考文献

[1]开放神经网络交换https://github.com/onnx/

[2]ONNXhttps://onnx.ai/

版本历史

在R2018a中引入