本文主要是介绍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 模型下载的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!