Deep Learning Toolbox Converter for PyTorch Models

Import pretrained PyTorch models into MATLAB
1,7K descargas
Actualizado 11 sep 2024
The converter for PyTorch models enables you to import pretrained PyTorch models and weights into MATLAB.
To import a PyTorch network in MATLAB, please refer to importNetworkFromPyTorch.
Note: the model must be traced in PyTorch before importing into MATLAB. See below for an example:
# This example loads a pretrained PyTorch model from torchvision,
# traces it with example inputs, and saves the trace as a .pt file.
import torch
from torchvision import models
# Load the model with pretrained weights
model = models.mobilenet_v2(pretrained=True)
# Call "eval" to ensure that layers like batch norm and dropout are set to
# inference mode
model.eval()
# Move the model to the CPU
model.to("cpu")
# Create example inputs
X = torch.rand(1, 3, 224, 224)
# Trace model with the example input
traced_model = torch.jit.trace(model.forward, X)
# Save the traced model to a .pt file
traced_model.save('traced_mobilenetv2.pt')
The initial release in R2022b supports importing image classification models. Support for other model types will be added in future updates.
Compatibilidad con la versión de MATLAB
Se creó con R2022b
Compatible con cualquier versión desde R2022b hasta R2024b
Compatibilidad con las plataformas
Windows macOS (Apple Silicon) macOS (Intel) Linux

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!