MATLAB的答案

Matlab ONNX CNN模型——推理图像分类结果

9的观点(30天)
安德里亚·道尔
安德里亚·道尔 2021年8月3日
回答: Sivylla Paraskevopoulou 2022年5月9日
你好所有的,
我使用的 exportONNXNetwork(净、文件名) 的函数导出一个微调CNN .mat模型(mobilenet pretrained和调整新数据集).onnx模型。
当使用.onnx模型在图像分类推理我得到错误的分类结果。我比较结果,我从Matlab与ONNX推理推理,它并不相同,虽然模型导出的,我检查网络的权重和偏见。
我需要做任何额外的转换吗?有两个模型之间的区别(我认为输入是形状在ONNX 3 * 224 * 224 224 * 224 * 3的Matlab) ?
提前谢谢你。感谢你的帮助!

在回答这个问题。

答案(2)

崔
崔 2021年8月4日
是的,你是正确的,在onnnx文件的输入是形状BatchSize * 3 * 224 * 224 224 * 224 * 3 * BatchSize在Matlab中,
1评论
安德里亚·道尔
安德里亚·道尔 2021年8月4日
我使用一个步骤纠正图像的形状,因为我没有得到任何错误时进行推理与.onnx模型预测和所有类概率方法不同于推理出正确的预测在Matlab完成。
我试图确保网络是正确的还从.mat传递到.onnx .oonx .mat模型和运行与新.mat推理模型和我得到正确的预测。
所以我没有得到问题:在哪里(

登录置评。

Sivylla Paraskevopoulou
Sivylla Paraskevopoulou 2022年5月9日
的 推理比较ONNX和进口网络图像分类 示例显示了如何比较图像分类结果之间ONNX模型和深度学习网络工具箱。

类别

找到更多的在识别、目标检测和语义分割在帮助中心和文件交换

标签

社区寻宝

找到宝藏在MATLAB中央,发现社区如何帮助你!

开始狩猎!