python加载resnet预训练模型

可以使用PyTorch框架中的torchvision库来加载预训练的ResNet模型。首先需要安装torchvision库：

pip install torchvision

然后在代码中加载模型：

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`。最后，我们打印输出结果。