58、resnet50 模型下载

本文主要是介绍58、resnet50 模型下载，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

上一节配置好 python 环境之后，本节开始下载 resnet50 训练好的模型。我们使用以下代码完成 resnet50 的模型下载。

import numpy as np
from torchvision import models
import torchresnet50 = torch.hub.load('pytorch/vision:v0.10.0', 'resnet50', pretrained=True)
resnet50.eval()
print(resnet50)

代码链接在：practice/model/resnet50_parser.py · iwaihou/cv_learning_from_scratch - Gitee.com

可以先看前 7 行，其余的暂时不需要看。

torch.hub.load

torch.hub.load 是 PyTorch 库中的一个函数，用于从 PyTorch Hub 中加载预训练的模型，示例中通过这个接口，下载了 resnet50 这个模型。

在这个接口中，pretrained = True 说明下载的模型是已经预训练好的，也就是模型中包含有训练好的参数。

resnet50.eval() 表示将模型设置为推理模式。所谓推理模式也就是使用该模型来进行推理而非训练。

这篇关于58、resnet50 模型下载的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---