MATLAB: Squeezenet (in this example) load to ONNX export and import does not work.

Question

If I import squeezenet, then export it as ONNX, then try to import it as ONNX, I get an error. It just seems that these 3 lines should just…work. Come on!

snet = squeezenet;
exportONNXNetwork(snet,'squeezenet.onnx');
snet_onnx = importONNXNetwork('squeezenet.onnx');

I get the error:

 Error using nnet.internal.cnn.onnx.importONNXNetwork>iValidateInputs (line 29)
 Expected OutputLayerType to be nonempty.
 Error in nnet.internal.cnn.onnx.importONNXNetwork (line 7)
 [Filename, OutputLayerType, UserImageInputSize, Classes] = iValidateInputs(filename, varargin{:});
 Error in importONNXNetwork (line 53)
 Network = nnet.internal.cnn.onnx.importONNXNetwork(ModelFile, varargin{:});

Answer 1

Thanks, that does it!

Adding

'OutputLayerType','classification'