Yolov5自定义对象检测模型未加载
Posted
技术标签:
【中文标题】Yolov5自定义对象检测模型未加载【英文标题】：Yolov5 custom object detection model not loading 【发布时间】：2021-09-29 17:14:44 【问题描述】：我已经使用 yolov5 为 4 个类训练了一个自定义对象检测模型。我已经下载了 best.pt 文件。我仍然对如何使用 pytorch 加载这个模型感到困惑。
我尝试按照 Yolov5 官方文档运行以下代码来加载模型
model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')
但是当我尝试打印 model.names 时，我没有得到我的自定义类名。有什么我想念的吗？
谢谢！
【问题讨论】：
【参考方案1】：-
尝试通过添加以下内容来清除缓存：
force_reload=True
将模型存储在名为 model 的本地文件夹中（如果您在本地
试图运行它）
然后尝试添加这个：
model = torch.hub.load('ultralytics/yolov5', 'custom', path='model/best-m.pt', force_reload=True)
【讨论】：
以上是关于Yolov5自定义对象检测模型未加载的主要内容，如果未能解决你的问题，请参考以下文章