2、PyTorch訓練YOLOv11—訓練篇(detect)—Windows系統

夕西行發表於2024-10-09

1、標註樣本

使用labelImg標註樣本,注意選擇YOLO模式

labelImg下載地址 github.com

2、安裝ultralytics包

先進入到自己建立的conda環境中,把包安裝到該環境裡

pip install ultralytics

如果已經有了ultralytics,可以更新最新版本

pip install -U ultralytics

最終 miniconda3\envs\你的環境名\Lib\site-packages 裡有個ultralytics資料夾

用pyCharm開啟這個資料夾,可以方便的檢視一些py檔案

3、組織樣本

以coco8為例,學習組織和訓練自己的樣本

3.1 下載coco8樣本

該檔案裡可以得到下載地址,直接複製連結用瀏覽器下載

download: https://github.com/ultralytics/assets/releases/download/v0.0.0/coco8.zip

解壓後放到D:/datasets中

把上圖的coco8.yaml也放到D:/datasets中,如下圖所示

修改coco8.yaml兩處

# 指定為自己資料集的路徑
path: D:/datasets/coco8 # dataset root dir
# 遮蔽最下邊的下載
# download: https://github.com/ultralytics/assets/releases/download/v0.0.0/coco8.zip 

3.2 pyCharm建立python專案

main.py內容如下

2.2 PyTorch訓練YOLO

官網 GitHub - ultralytics/ultralytics: NEW - YOLOv8 🚀 in PyTorch > ONNX > OpenVINO > CoreML > TFLite

參考 手把手教你使用YOLOv11訓練自己資料集(含環境搭建 、資料集查詢、模型訓練)-CSDN部落格

從官網下載pt模型

【其他】

VS使用conda虛擬環境,管理 Python 環境和直譯器 - Visual Studio (Windows) | Microsoft Learn

相關文章