python加载resnet预训练模型
时间: 2023-10-14 11:25:08 浏览: 95
预训练的Pytorch人脸检测(MTCNN)和识别(InceptionResnet)模型-Python开发
可以使用PyTorch框架中的torchvision库来加载预训练的ResNet模型。首先需要安装torchvision库:
```
pip install torchvision
```
然后在代码中加载模型:
```python
import torch
import torchvision.models as models
# 加载ResNet50模型
model = models.resnet50(pretrained=True)
# 使用模型进行推理
input_tensor = torch.randn(1, 3, 224, 224)
output = model(input_tensor)
print(output)
```
在上述代码中,我们首先导入了PyTorch和torchvision库,然后使用`models.resnet50(pretrained=True)`加载预训练的ResNet50模型。接着,我们使用模型进行推理,随机生成一个大小为1x3x224x224的输入张量,通过模型进行前向计算得到输出结果`output`。最后,我们打印输出结果。
阅读全文