导入和导出ONNX™ (MATLAB中的开放式神经网络交换模型,用于与其他深度学习框架的互操作性。ONNX使模型能够在一个框架中进行训练,并转移到另一个框架中进行推理。
从您的操作系统或在MATLAB中打开onnxconverter.mlpkginstall文件将启动您的版本的安装过程。
此mlpkginstall文件适用于R2018a及更高版本。
用法示例:
%%导出为ONNX模型格式
净=挤压净;%待导出的预训练模型
filename='squeezenet.onnx';
exportONNXNetwork(net,文件名);
%%导入导出的网络
net2=importONNXNetwork('squeezenet.onnx','OutputLayerType','classification');
%比较两个网络对随机输入图像的预测
img=rand(net.Layers(1).InputSize);
y=预测(净,img);
y2=预测(net2,img);
最大值(abs(y-y2))
要在MATLAB中导入ONNX网络,请参考:
//www.tatmou.com/help/deeplearning/ref/importonnxnetwork.html
要从MATLAB导出ONNX网络,请参考:
//www.tatmou.com/help/nnet/ref/exportonnxnetwork.html