我已经为4个类使用yolov5训练了一个自定义对象检测模型。我已经下载了最好的.pt文件。我仍然对如何使用pytorch加载这个模型感到困惑。我试着按照Yolov5官方文档运行以下代码来加载模型model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')但是当我尝试打印model.names时，我没有得到自定义类名。我缺什么了吗？非常感谢。
- 尝试通过添加以下内容来清除缓存：
force_reload=True - 将模型存储在名为model的本地文件夹中(以防您在本地尝试运行此(
然后尝试添加此：
model = torch.hub.load('ultralytics/yolov5', 'custom', path='model/best-m.pt', force_reload=True)