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