Yolov5 自定義對象檢測模型未加載

Question

我使用 yolov5 為 4 個類訓練了一個自定義對象檢測模型。 我已經下載了 best.pt 文件。 我仍然對如何使用 pytorch 加載這個模型感到困惑。 我嘗試按照 Yolov5 官方文檔model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')運行以下代碼來加載模型，但是當我嘗試打印 model.names 時，我沒有得到我的自定義類名。 有什么我想念的嗎？ 謝謝！

Answer 1

1. 嘗試通過添加以下內容來清除緩存： force_reload=True
2. 將您的模型存儲在名為 model 的本地文件夾中（以防您在本地嘗試運行它）

然后嘗試添加這個：

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