Main Content

本页翻译不是最新的。点击此处可查看最新英文版本。

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

从外部深度学习平台导入预训练网络

从 TensorFlow™ 2、TensorFlow-Keras、PyTorch®、ONNX™(开放式神经网络交换)模型格式和 Caffe 中导入网络和层图。有关详细信息,请参阅预训练的深度神经网络Interoperability Between Deep Learning Toolbox, TensorFlow, PyTorch, and ONNX

您必须有支持包才能在 Deep Learning Toolbox™ 中运行导入函数。如果未安装支持包,则每个函数都会在附加功能资源管理器中提供对应支持包的下载链接。建议将支持包下载到您正在运行的 MATLAB® 版本的默认位置。您也可以从以下链接直接下载支持包。

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

函数

全部展开

TensorFlow 导入

importNetworkFromTensorFlowImport TensorFlow network as MATLAB network (自 R2023b 起)
importTensorFlowNetwork(To be removed) Import pretrained TensorFlow network (自 R2021a 起)
importTensorFlowLayers(To be removed) Import layers from TensorFlow network (自 R2021a 起)
importKerasNetwork(To be removed) Import pretrained Keras network and weights
importKerasLayers(To be removed) Import layers from Keras network

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 起)
importONNXNetwork(To be removed) Import pretrained ONNX network
importONNXLayers(To be removed) Import layers from ONNX network
importONNXFunctionImport pretrained ONNX network as a function (自 R2020b 起)

importONNXFunction 导入的参数

ONNXParametersParameters of imported ONNX network for deep learning (自 R2020b 起)
freezeParametersConvert learnable network parameters in ONNXParameters to nonlearnable (自 R2020b 起)
unfreezeParametersConvert nonlearnable network parameters in ONNXParameters to learnable (自 R2020b 起)
addParameterAdd parameter to ONNXParameters object (自 R2020b 起)
removeParameterRemove parameter from ONNXParameters object (自 R2020b 起)
functionLayerFunction layer (自 R2021b 起)
findPlaceholderLayersFind placeholder layers in network architecture imported from Keras or ONNX
replaceLayerReplace layer in neural network
assembleNetwork(Not recommended) Assemble deep learning network from pretrained layers
PlaceholderLayerLayer replacing an unsupported Keras or ONNX layer
addLayersAdd layers to neural network
removeLayersRemove layers from neural network

主题

导入

自定义层