开始使用免费开始使用

加载预训练模型

您正在构建一个用于给社交媒体图片打标签的应用。此任务需要高准确率和高速度。您将使用预训练的 ResNet18 模型来推断图像类别。

本练习是课程的一部分

使用 PyTorch 进行图像深度学习

查看课程

练习说明

  • torchvision.models 导入 resnet18ResNet18_Weights
  • 使用 resnet18() 实例化 model,并将 weights 参数设置为 weights
  • model 设置为评估模式。
  • 初始化输入变换,并将其赋值给 transform

交互式实操练习

通过完成这段示例代码来试试这个练习。

# Import resnet18 model
from torchvision.models import ____

# Initialize model with default weights
weights = ResNet18_Weights.DEFAULT
model = ____

# Set model to evaluation mode
model.____

# Initialize the transforms
transform = ____
编辑并运行代码