我能夠在具有 Internet 連接的 PC 上使用 yolov5 運行 Flask 應用程式。我按照 yolov5 檔案中提到的步驟并使用了這個檔案:yolov5/utils/flask_rest_api/restapi.py,
但我需要實作相同的離線(在特定的 PC 上)。現在的問題是,當我使用以下內容時:
model = torch.hub.load("ultralytics/yolov5", "yolov5", force_reload=True)
它試圖從互聯網上下載模型。并拋出錯誤。
Urllib.error.URLError: <urlopen error [Errno - 2] name or service not known>
如何離線獲得相同的結果。
提前致謝。
uj5u.com熱心網友回復:
如果要離線運行檢測,則需要已下載模型。

uj5u.com熱心網友回復:
正如@Milan Hlinak 所述,它解決了我的問題。
進一步如https://docs.ultralytics.com/tutorials/pytorch-hub/所述,此代碼可以訪問除在 PyTorch-model-hub 上傳的模型之外的任何自定義訓練模型。
path_hubconfig = 'absolute/path/to/yolov5'
path_trained_model = 'absolute/path/to/best.pt'
model = torch.hub.load(path_hubconfig, 'custom', path=path_trained_model, source='local') # local repo
使用此代碼,物件檢測由本地保存的自定義訓練模型執行。一次,自定義訓練模型保存在本地,這段代碼直接訪問它,避免了互聯網的任何必要性。
轉載請註明出處,本文鏈接:https://www.uj5u.com/caozuo/433483.html
標籤:python-3.x 烧瓶 火炬 约洛夫5
