加载预训练模型
您正在构建一个用于给社交媒体图片打标签的应用。此任务需要高准确率和高速度。您将使用预训练的 ResNet18 模型来推断图像类别。
本练习是课程的一部分
使用 PyTorch 进行图像深度学习
练习说明
- 从
torchvision.models导入resnet18和ResNet18_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 = ____