How do I export a neural network from MATLAB?

I have a neural network which I trained using MATLAB. I want to export the network so I can use it with other frameworks, for example Caffe. How do I do that?

Answer 1

[MathWorks Support Team]

You can export a Deep Learning Toolbox network or layer graph to TensorFlow and ONNX using the "exportNetworkToTensorFlow" and "exportONNXNetwork" functions, respectively.  For more information about exporting networks to external deep learning platforms, please see the following documentation page:

https://www.mathworks.com/help/releases/R2023b/deeplearning/export-deep-neural-networks.html

Alternatively, you could export via the MATLAB Compiler SDK.

Using the MATLAB Compiler SDK, you can save the trained network as a MAT file, and write a MATLAB function that loads the network from the file, performs the desired computation, and returns the network's output.

You can then compile your MATLAB function into a shared library to be used in your C/C++, .NET, Java, or Python project.

You can find more information about MATLAB Compiler SDK in the following link:

https://www.mathworks.com/help/releases/R2023b/compiler_sdk/ml_code/librarycompiler-app.html

Furthermore, the objects that MATLAB uses to represent neural networks are transparent, and you can therefore access all the information that describes your trained network.

For example, you will get an object of type "SeriesNetwork", which is a trained Convolutional Neural Network. You can then see the weights and biases of the trained network:

convnet.Layers(2).Weights
convnet.Layers(2).Bias

Then, using for example Caffe's MATLAB interface, you should be able to save a Convolutional Neural Network as a Caffe model. The code for the MATLAB interface is in the following link:

https://github.com/BVLC/caffe/tree/master/matlab

and includes a classification demo that shows you how to use the interface.

Please note that the above code is not developed or supported by MathWorks Technical Support. If you have any questions about how to use the code, please contact the project's developers.

[michael scheinfeild]

should i compile to full caffe in c++ ?

Answer 2

[Maria Duarte Rosa]

The exportONNXNetwork function in Deep Learning Toolbox Converter for ONNX Model Format allows one to export a trained deep learning network to the ONNX™ (Open Neural Network Exchange) model format. The ONNX model can then be imported into other deep learning frameworks that support ONNX model import.

To use the network with TensorFlow, use the exportNetworkToTensorFlow function that is part of the Deep Learning Toolbox Converter for TensorFlow Models.

[michael scheinfeild]

still i have no success to import it to c++ from onnx there are many issues of compilation

Answer 3

[michael scheinfeild]

after testing onnx i found that the output of convolutions is not the same as in matlab .