Yolov5自定义对象检测模型未加载答案

【问题标题】：Yolov5 custom object detection model not loadingYolov5自定义对象检测模型未加载
【发布时间】：2021-09-29 17:14:44
【问题描述】：

我已经使用 yolov5 为 4 个类训练了一个自定义对象检测模型。我已经下载了 best.pt 文件。我仍然对如何使用 pytorch 加载这个模型感到困惑。我尝试按照 Yolov5 官方文档运行以下代码来加载模型 model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt') 但是当我尝试打印 model.names 时，我没有得到我的自定义类名。有什么我想念的吗？谢谢！

【问题讨论】：

标签： opencv object-detection yolov5

【解决方案1】：

尝试通过添加以下内容来清除缓存：force_reload=True
将模型存储在名为 model 的本地文件夹中（如果您在本地试图运行它）

然后尝试添加这个：

model = torch.hub.load('ultralytics/yolov5', 'custom', path='model/best-m.pt', force_reload=True)

【讨论】：