主要内容

来自外部平台的预训练网络

R2026b
使用来自外部深度学习平台的预训练网络

使用来自外部深度学习平台的预训练网络有两种方法:导入模型;使用协同执行来训练和测试外部模型。在可能的情况下,首选方法是导入模型。

Deep Learning Toolbox™ 支持从 TensorFlow™ 2、TensorFlow-Keras、Keras 3、PyTorch®、ONNX™(开放式神经网络交换)模型格式和 Caffe 导入神经网络。您可以使用深度网络设计器以交互方式导入网络,也可以使用命令行函数以编程方式导入网络。该 App 会生成一个导入报告,总结导入过程,并突出显示任何需要采取的操作。有关详细信息,请参阅预训练的深度神经网络和Interoperability Between Deep Learning Toolbox, TensorFlow, PyTorch, and ONNX。

Deep Learning Toolbox 中的导入函数需要特定的支持包。如果未安装所需的支持包,则每个函数都会在附加功能资源管理器中提供对应支持包的下载链接。请将支持包下载到当前使用的 MATLAB® 版本的默认位置。支持包也可以直接下载:

MATLAB 与外部深度学习平台之间的互操作

如果无法导入预训练网络,您可以使用协同执行来运行外部模型。基于 Python® 的模型支持协同执行,包括 TensorFlow、ONNX、PyTorch 和自定义 Python 模型。对于 PyTorch 模型,您可以使用 PyTorchModel 对象直接在 MATLAB 中运行模型。此对象充当包装器,通过 MATLAB–Python 接口在 Python 中执行模型,从而无需导入模型即可实现集成。您还可以使用 arrayToTorchTensor 函数和 torchTensorToArray 函数在 MATLAB 数值数组与 PyTorch 张量之间进行转换。

App

深度网络设计器设计和可视化深度学习网络

函数

全部展开

TensorFlow 导入

importNetworkFromTensorFlowImport TensorFlow network as MATLAB network (自 R2023b 起)
importNetworkFromKerasImport Keras 3 network as MATLAB network (自 R2026a 起)

PyTorch 导入

importNetworkFromPyTorchImport PyTorch network as MATLAB network (自 R2022b 起)

Caffe 导入

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

ONNX 导入

importNetworkFromONNXImport ONNX network as MATLAB network (自 R2023b 起)

由 importONNXFunction 导入的参数

ONNXParametersParameters of imported ONNX network for deep learning
freezeParametersConvert learnable network parameters in ONNXParameters to nonlearnable
unfreezeParametersConvert nonlearnable network parameters in ONNXParameters to learnable
addParameterAdd parameter to ONNXParameters object
removeParameterRemove parameter from ONNXParameters object
dlnetwork深度学习神经网络
functionLayerFunction layer
replaceLayerReplace layer in neural network
addLayers向神经网络添加层
removeLayers从神经网络中删除层
arrayToTorchTensorConvert MATLAB numeric array to PyTorch Tensor (自 R2026b 起)
torchTensorToArrayConvert PyTorch Tensor to MATLAB numeric array (自 R2026b 起)
forwardCompute PyTorchModel output by invoking Python model on input (自 R2026b 起)
addFunctionConfigure data transfer settings for Python method or function of PyTorchModel (自 R2026b 起)
callFunctionCall method of underlying Python PyTorch model instance or function taking model as its first argument (自 R2026b 起)
exportSave underlying PyTorch model in different PyTorch file format (自 R2026b 起)
reloadReload PyTorch model from stored model loading properties (自 R2026b 起)

对象

PyTorchModelReference to a PyTorch model for Python execution (自 R2026b 起)

主题

导入

Simulink 模块的 Python 协同执行

MATLAB 命令行中的 PyTorch 协同执行

自定义层