Matlab CNN model to ONNX - Inference Image Classification Results
4 次查看(过去 30 天)
显示 更早的评论
Hello all,
I used 'exportONNXNetwork(net,filename)' function to export a fine-tuned CNN .mat model (mobilenet pretrained and fine-tuned on new dataset) to .onnx model.
When using the .onnx model in image classification inference I am getting wrong classification results. I compared the results that I am getting from Matlab inference with ONNX inference, it is not the same although the model was well exported and I checked some of the weights and biases of the network.
Does I need to make any additional conversion ? Is there any difference between the two models (I think the input is of shape 3*224*224 in ONNX instead of 224*224*3 in Matlab) ?
Thank you in advance. Appreciate your help!
0 个评论
回答(2 个)
cui,xingxing
2021-8-4
yes , you are correct, in onnnx file the input is of shape BatchSize*3*224*224 instead of 224*224*3*BatchSize in Matlab,
Sivylla Paraskevopoulou
2022-5-9
The Inference Comparison Between ONNX and Imported Networks for Image Classification example shows how to compare image classification results between an ONNX model and a Deep Learning Toolbox network.
0 个评论
另请参阅
类别
在 Help Center 和 File Exchange 中查找有关 Image Data Workflows 的更多信息
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!