Matlab CNN model to ONNX - Inference Image Classification Results

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!

[PARVEZ]

Same problem for me too what to do to correct this problem please help

Answer 1

[xingxingcui]

yes , you are correct, in onnnx file the input is of shape BatchSize*3*224*224 instead of 224*224*3*BatchSize in Matlab,

[Andrea Daou]

I used a step to correct the image shape because I am not getting any error when doing the inference with the .onnx model but the prediction and all the classes probabilities are way different from the inference done in Matlab that gives correct predictions.

I tried to make sure that the network is correctly passing from .mat to .onnx by repassing from the .oonx model to .mat and running the inference with the new .mat model and I got right predictions.

So I am not getting where's the problem :(

Answer 2

[Sivylla Paraskevopoulou]

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.