58、resnet50 模型下载

2024-01-07 18:53:51

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

import numpy as np
from torchvision import models
import torch

resnet50 = 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() 表示将模型设置为推理模式。所谓推理模式也就是使用该模型来进行推理而非训练。

文章来源:https://blog.csdn.net/dongtuoc/article/details/135436482
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。