How to load a fully connected Pytorch model (trained.model) into matlab ?

Question

Oualid Doukhi 2020-10-8

1
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/607941-how-to-load-a-fully-connected-pytorch-model-trained-model-into-matlab

评论： Sivylla Paraskevopoulou 2022-12-13

I have i fully connected neural networks which was trained in pytorch, the model was saved as (.model) i would like to load this model to matlab is there any way how to di it?

1 个评论
显示 -1更早的评论隐藏 -1更早的评论

QUAN WANG 2022-11-12

Hello, have you solved the issue ?

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Ameer Hamza 2020-10-8

4
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/607941-how-to-load-a-fully-connected-pytorch-model-trained-model-into-matlab#answer_508311

Yes, the deep learning toolbox supports Framework Interoperability: https://www.mathworks.com/products/deep-learning.html#frm. It directly supports importing models from TensorFlow and Caffe. For PyTorch, you might need to use the ONNX format to load it in MATLAB: https://www.mathworks.com/help/deeplearning/ref/importonnxnetwork.html. This webpage shows how to convert PyTorch to ONNX: https://pytorch.org/tutorials/advanced/super_resolution_with_onnxruntime.html.

4 个评论
显示 2更早的评论隐藏 2更早的评论

Yam Alcaraz 2021-3-22

Just to help out, I tried it with pytorch and it works:

imported alexnet from torchvision
exported as an .onnx file
imported in matlab as importONNXNetwork('alexnet.onnx','OutputLayerType','classification');

Med Future 2022-3-11

@Yam Alcaraz Can you share your code for both pytorch and MATLAB

请先登录，再进行评论。

Answer 2

Sivylla Paraskevopoulou 2022-10-7

1
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/607941-how-to-load-a-fully-connected-pytorch-model-trained-model-into-matlab#answer_1068905

In R2022b we introduced the Deep Learning Toolbox Converter for PyTorch Models support package. This initial release supports importing image classification models. Support for other model types will be added in future releases. Use the function importNetworkFromPyTorch to import a PyTorch model. Make sure that the PyTorch model that you are importing is pretrained and traced.

For more details, check out the blog post What’s New in Interoperability with TensorFlow and PyTorch and the importNetworkFromPyTorch documentation page.

If you want to import another type (not image classification) model from PyTorch, convert you model to the ONNX model format and then, use the importONNXNetwork function.

4 个评论
显示 2更早的评论隐藏 2更早的评论

Jean-Robert Philippe 2022-12-13

In the opposite direction, how can we export a .pt model from Matlab? I obtained very good results with a model pre-trained in Matlab, but this kind of model doesn't exist in PyTorch.

I tried onnx format and onnx2pytorch, but it doesn't work.

Sivylla Paraskevopoulou 2022-12-13

Glad to hear you obtained good results with MATLAB!

At the moment, there is no direct path to convert MATLAB models to PyTorch models. You have to do it via ONNX by using the exportONNXNetwork function.

If exporting to TensorFlow is suitable for your workflow, you can use the exportNetworkToTensorFlow function to export MATLAB models to TensorFlow.

请先登录，再进行评论。