Main Content

Pretrained Networks from External Platforms

Import pretrained networks from external deep learning platforms

Import neural networks from TensorFlow™ 2, TensorFlow-Keras, PyTorch®, the ONNX™ (Open Neural Network Exchange) model format, and Caffe. For more information, see Pretrained Deep Neural Networks and Interoperability Between Deep Learning Toolbox, TensorFlow, PyTorch, and ONNX.

You must have support packages to run the import functions in Deep Learning Toolbox™. If the support package is not installed, each function provides a download link to the corresponding support package in the Add-On Explorer. A recommended practice is to download the support package to the default location for the version of MATLAB® you are running. You can also directly download the support packages from the following links.

Diagram showing the interoperability between Deep Learning Toolbox, TensorFlow, ONNX, and PyTorch.

Functions

expand all

TensorFlow Import

importNetworkFromTensorFlowImport TensorFlow network as MATLAB network (Since R2023b)

PyTorch Import

importNetworkFromPyTorchImport PyTorch network as MATLAB network (Since R2022b)

Caffe Import

importCaffeNetworkImport pretrained convolutional neural network models from Caffe
importCaffeLayersImport convolutional neural network layers from Caffe

ONNX Import

importNetworkFromONNXImport ONNX network as MATLAB network (Since R2023b)

Parameters Imported by importONNXFunction

ONNXParametersParameters of imported ONNX network for deep learning (Since R2020b)
freezeParametersConvert learnable network parameters in ONNXParameters to nonlearnable (Since R2020b)
unfreezeParametersConvert nonlearnable network parameters in ONNXParameters to learnable (Since R2020b)
addParameterAdd parameter to ONNXParameters object (Since R2020b)
removeParameterRemove parameter from ONNXParameters object (Since R2020b)
dlnetworkDeep learning neural network (Since R2019b)
functionLayerFunction layer (Since R2021b)
replaceLayerReplace layer in neural network
addLayersAdd layers to neural network
removeLayersRemove layers from neural network

Topics

Import

Python Coexecution

Custom Layers