yolo26n.pt 模型可用來偵測圖片,這個模型包含了網路架構及訓練後的權重,此模型又稱為預訓練權重。
預設權重使用微軟的 COCO 資料集訓練而成的。COCO 收集了16萬4000張的圖片,共有 80 種物件。
yaml 及 pt
yolo26n.yaml 只是模型的設計圖(網路結構),沒有任何訓練資料;而 yolo26n.pt 則是已經訓練完成的模型,裡面包含結構和權重。換句話說,yolo26n.pt 就是用 yolo26n.yaml 這個架構訓練出來的成果。
安裝套件
請記得先安裝 Pytorch for cuda,然後才安裝 ultralytics。
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu130 --no-cache-dir
pip install ultralytics labelimg
COCO 資料集下載
COCO 資料集可以由如下網址下載
