model=torch.hub.load加载本地的best.pt

于 2024-05-23 16:58:29 发布
阅读量512 收藏 1
点赞数 1
文章标签： yolov5 pytorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/XUQIAN_LIUQIAN/article/details/139151731
版权

要调用best.pt的项目路径：

当前项目的路径如下：

        我要在yolov5-D435i-main项目中去调用yolov5-6.0项目中训练好的best.pt文件，然后得把yolov5-6.0项中你要用到的best.pt文件放在yolov5-D435i-main项目中。

model = torch.hub.load('E:/yolo/yolov5-6.0', 'custom', path=r'best.pt', source='local')  # local repo
model.conf = 0.8  # set confidence threshold

        如此，成功！！！

星河@星辰大海
关注
YOLOv5的Tricks | 【Trick10】从PyTorch Hub加载YOLOv5
Clichong
08-06 2330
如有错误，恳请指出。严格意义来说，这篇文章算不上是yolov5的一个训练技巧，而且yolov5的训练技巧在专栏之前的文章中基本已经都讲完了，之后可以会补充一些yolov5项目的其他内容。比如这篇，从 加载YOLOv5模型，不需要克隆 不过很多朋友（包括我），在整理之前，还不知道pytorch hub是什么，所以这里顺便介绍一些pytorch hub，然后再根据官方链接介绍如何不需要克隆就可以玩转yolov5项目。官方介绍：Towards Reproducible Research with PyTorch
机器学习笔记 - 基于Torch Hub和YOLOv5和SSD的目标检测
学以致用 知行合一
05-11 1995
对象检测是一个很有意思的领域。让机器识别图像中物体的确切位置有各种各样的关键用途。从人脸检测系统到帮助自动驾驶汽车安全导航，不胜枚举。 YOLO 或“ You Only Look Once ”（2015 年）引入了一种巧妙的方法来解决简单CNN检测器的缺点。我们将每个图像分割成一个S × S网格，得到每个单元格对应的对象位置。当然，有些单元格不会有任何对象，而有些单元格会出现在多个单元格中，见下图。
【NLP】torch hub工具的使用：torch.hub.load、pytorch预训练模型加载
sunshine77_的博客
08-17 4301
2.4 加载和使用预训练模型 学习目标: 了解加载和使用预训练模型的工具. 掌握加载和使用预训练模型的过程. 加载和使用预训练模型的工具: 在这里我们使用torch.hub工具进行模型的加载和使用. 这些预训练模型由世界先进的NLP研发团队huggingface提供. 加载和使用预训练模型的步骤: 第一步: 确定需要加载的预训练模型并安装依赖包. 第二步: 加载预训练模型的映射器tokenizer. 第三步: 加载带/不带头的预训练模型. 第四步: 使用模型获得输出结果. 第一步: 确..
pytorch实现从本地加载 .pth 格式模型
12-20
可以从官网加载预训练好的模型： import torchvision.models as models model = models.vgg16(pretrained = True) print(model) 但是经常会出现因为下载速度太慢而出现requests.exceptions.ConnectionError: (‘Connection aborted.’, TimeoutError(10060, ‘由于连接方在一段时间后没有正确答复或连接的主机没有反应，连接尝试失败。’, None, 10060, None))这种错误，因此需要我们手动去下载 .pth 文件（百度云也很慢，如果你是S
torch.hub.load 加载本地模型（已解决）
michael_quan2013的博客
03-29 1万+
不让 torch.hub.load 联网下载模型，改为 torch.hub.load 加载本地模型。本地模型默认的下载路径是：/root/.cache/torch/hub……运行网上的项目，有时会卡住或者超时，原因是 torch.hub.load 默认会去网上找模型，有时会因为网络问题而报错。
torch.hub.load()把联网加载修改为本地加载模型
qq_51248362的博客
10-04 6351
我们常用torch.load导入本地模型，但是最近在一个yolov5+gradio的项目中遇到了torch.hub.load方法，这是一个需要联网从GitHub或其他远程位置加载预训练模型的方法。但是联网从github加载通常会出现连接超时的情况，因此转为从本地加载会是更好的选择。
torch.hub 加载yolov5本地（自）训练模型
Major_S的博客
09-22 2858
【代码】torch.hub 加载yolov5本地（自）训练模型。
torch.hub.load()函数的使用——联网加载权重以及如何加载本地权重
qq_37346140的博客
10-20 1万+
torch.hub.load()函数的使用——联网加载权重以及如何加载本地权重，参数设置
mode1 = torch.hub.load("./","custom",path="runs/train/exp6/weights/best.pt",source="local")
最新发布
06-18
1. `torch.hub.load("./", "custom")`: 这行代码是从本地文件路径 "./" 加载名为 "custom" 的预定义模型。"custom" 可能是一个模型的名字或者存储模型的模块名称。 2. `path="runs/train/exp6/weights/best.pt"`: ...
import cv2 import torch # 加载模型 model = torch.hub.load('./base_s16', 'custom' , path='base_s16/weights/best.pt', source='local') # 1.获取视频对象 cap = cv2.VideoCapture('gao.mp4') # 2.判断是否读取成功 while(cap.isOpened()): # 3.获取每一帧图像 ret, frame = cap.read() # 镜像 # frame = cv2.flip(frame, 1) # print(model) # bgr --> rgb img_cvt = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) results = model(img_cvt) # 推理结果 pd = results.pandas().xyxy[0] person_list = pd[pd['name'] == 'person'].to_numpy() print(person_list) helmet_list = pd[pd['name'].str.contains('helmet')].to_numpy() print(helmet_list) # 4.获取成功显示图像 if ret == True: cv2.imshow('frame',frame) # 5.每一帧间隔为25ms break # if cv2.waitKey(25) & 0xFF == ord('q'): # break # 6.释放视频对象 cap.release() cv2.destoryAllwindows()
06-10
model = torch.hub.load('./base_s16', 'custom', path='base_s16/weights/best.pt', source='local') # 获取视频对象 cap = cv2.VideoCapture('gao.mp4') # 判断是否读取成功 while cap.isOpened(): # 获取每一...
解决在使用torch.hub.load方法加载本地yolov5模型时的报错：custom() got an unexpected keyword argument ‘path‘
哦牛逼，我放我GitHub博客链接还报违规的
04-08 1979
非常申必的错误原因，解决方法是把 ‘path=’ 给删掉，得到以下新的语句。报了标题里的那个错误。更加申必的是百度居然找不到这个问题。参数位置不知道能不能改，建议别改。
torch hub工具的使用：torch.hub.load、pytorch预训练模型加载
qq_45990036的博客
10-19 2011
xuexi
torch.hub.load(‘pytorch/vision‘, ‘deeplabv3_resnet101‘, pretrained=True)运行方法
Lucky_Ape的博客
09-05 1902
首先下载 vision-master.zip torch_1.7.1_gpu) E:\PythonPRO\PytorchPro\G5>pip install vision-master.zip 实现上安装到 C:\Users\Administrator.cache\torch\hub (每个人安装的地方不一样，通过运行程序可以看出在什么地方) 下面主要用到2个文件 D:\ProgramData\Anaconda3\envs\pytorch_gpu\Lib\site-packages\tor.
torch.hub.load()解析，如何加载本地权重
热门推荐
qq_34317565的博客
01-04 3万+
用yolov5训练了一个权重，项目只能部署在本地 官方文档 torch.hub.load(repo_or_dir, model, *args, source='github', force_reload=False, verbose=True, skip_validation=False, **kwargs) [来源] 从 github 存储库或本地目录加载模型。 注意：加载模型是典型的用例，但这也可用于加载其他对象，例如分词器、损失函数等。 如果source是“github”，rep
解决torch.hub.load加载网络模型异常
十年以上架构设计经验，专注于软件架构和人工智能领域，对机器视觉、NLP、音视频等领域都有涉猎
11-06 3234
torch.hub.load是一个强大的工具，为PyTorch用户提供了快速访问和使用预训练模型的能力。它简化了模型加载过程，使得用户可以更加专注于模型的应用和开发，而不是从头开始训练模型。
使用 PyTorch Hub 加载 YOLOv5
m0_68220374的博客
05-03 1667
本文记录了使用 PyTorch Hub 加载 YOLOv5的一些操作和指令
pytorch中的torch.hub.load()：以vggish为例
zly_Always_be的博客
03-26 1588
pytorch中的torch.hub.load()：加载预训练模型，以vggish为例
torch.hub.load自动下载预训练模型文件用以加载预训练模型
あずにゃん梓喵的博客
07-26 6623
日萌社 人工智能AI：Keras PyTorch MXNet TensorFlow PaddlePaddle 深度学习实战（不定时更新） 确定需要加载的预训练模型并安装依赖包 能够加载哪些模型可以参考NLP中的常用预训练模型 这里假设我们处理的是中文文本任务, 需要加载的模型是BERT的中文模型: bert-base-chinese 在使用工具加载模型前需要安...
【NLP】torch hub工具的使用：torch.hub.load、pytorch预训练模型加载、
knocky的博客
04-27 1万+
2.4 加载和使用预训练模型 学习目标: 了解加载和使用预训练模型的工具. 掌握加载和使用预训练模型的过程. 加载和使用预训练模型的工具: 在这里我们使用torch.hub工具进行模型的加载和使用. 这些预训练模型由世界先进的NLP研发团队huggingface提供. 加载和使用预训练模型的步骤: 第一步: 确定需要加载的预训练模型并安装依赖包. 第二步: 加载预训练模...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值