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
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。 如若内容造成侵权/违法违规/事实不符,请联系我的编程经验分享网邮箱:veading@qq.com进行投诉反馈,一经查实,立即删除!
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。 如若内容造成侵权/违法违规/事实不符,请联系我的编程经验分享网邮箱:veading@qq.com进行投诉反馈,一经查实,立即删除!