MATLAB的答案

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

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

在回答这个问题。

答案(1)

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

登录置评。