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

Question

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

Answer 1

模型尝试通过添加以下内容来清除缓存:将模型放在名为

1. 的本地文件夹中(如果您在本地尝试运行此操作)

然后尝试添加此

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