本文介绍了在Yolov5-v7中如何解决自定义模型加载的问题，指出直接使用torch.load()加载不适用于Yolo训练的模型。作者分享了在GitHub issue中找到的解决方案，强调在hub.load()中设置custom参数可以实现本地自定义模型的加载，并且提供了使用PIL读取图片进行检测的方法。
@Author：Runsen
本想找下，有没有代码可以实现自定义模型的检测，发现百度没有一篇。
大家肯定说，直接用load加载不就是行了吗，别以为用torch.load(“”) 直接加载就Ok？ torch.load 加载是你通过torch训练的，不是yolo训练，那肯定不同
可怜的我去GitHub的issue找到了答案。
官方模型加载yolov5预训练模型是
超级会员免费看
import torch
# Model 下载
model = torch.hub.load('ultralytics/yolov5', 'yolov5s') # or yolov5n - yolov5x6, custom
3318
738
6172
1794
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
订阅专栏 解锁全文
扫一扫
