时间戳

首页 > 解决方案 > Yolov5自定义对象检测模型未加载

opencv - Yolov5自定义对象检测模型未加载

问题描述

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

标签： opencvobject-detectionyolov5

解决方案

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

然后尝试添加：

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

推荐阅读