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.
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) LinuxEtiquetas
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!Descubra Live Editor
Cree scripts con código, salida y texto formateado en un documento ejecutable.