如何导入resnet18的预训练模型

要导入ResNet-18的预训练模型，你可以使用PyTorch提供的torchvision库。首先，你需要安装PyTorch和torchvision库。然后，按照以下步骤导入ResNet-18的预训练模型：

1. 导入所需的库：

import torch
import torchvision.models as models

2. 创建ResNet-18模型的实例：

resnet = models.resnet18(pretrained=True)

这将创建一个ResNet-18模型的实例，并加载预训练的权重。

3. 使用模型进行推理或微调：
你可以使用导入的模型进行图像分类等任务。例如，对于图像分类任务，你可以使用以下代码：

# 假设有一张图像img
output = resnet(img)

这将返回模型对图像的分类结果。