图像缩略图

ONNX模型格式的深入学习工具箱转换器

导入和导出ONNX™ MATLAB中与其他深度学习框架互操作的模型

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兼容
平台兼容性
窗户 马科斯 Linux

社区寻宝

在MATLAB Central中查找宝藏,了解社区如何帮助您!

开始打猎吧!