2.1K下载
更新2021年9月22日
导入和导出ONNX™ (开放式神经网络交换)MATLAB中的模型,用于与其他深度学习框架的互操作性。ONNX使模型能够在一个框架中进行训练,并转移到另一个框架进行推理。
从操作系统或MATLAB中打开onnxconverter.mlpkginstall文件将启动已发布版本的安装过程。
此mlpkginstall文件适用于R2018a及更高版本。
用法示例:
%%导出为ONNX模型格式
净=挤压净;%要导出的预训练模型
文件名='squezenet.onnx';
exportONNXNetwork(net,文件名);
%%导入导出的网络
net2=importonxnetwork('squeezenet.onnx','OutputLayerType','classification');
%在随机输入图像上比较两个网络的预测
img=rand(净层(1)输入大小);
y=预测值(净,img);
y2=预测(net2,img);
最大值(绝对值(y-y2))
要在MATLAB中导入ONNX网络,请参考:
//www.tianjin-qmedu.com/help/deeplearning/ref/importonnxnetwork.html
要从MATLAB导出ONNX网络,请参考:
//www.tianjin-qmedu.com/help/nnet/ref/exportonnxnetwork.html
MATLAB版本兼容性
使用R2018a创建
与R2018a至R2021b兼容