YOLOV8 OBB 海思3516訓練流程
目錄
1、 下載帶GPU版本的torch(可選) 1
2、 安裝 ultralytics 2
3、 下載pycharm 社區版 2
4、安裝pycharm 3
5、新建pycharm 工程 3
6、 添加conda 環境 4
7、 訓練代碼 5
9、配置Ymal 文件 6
10、修改網絡結構 9
11、運行train.py 開始訓練模型 10
12、檢查數據是否標注正確 11
1、下載帶GPU版本的torch(可選)
在pytorch官網:
https://pytorch.org/get-started/locally/
下載GPU版本的torch。
`pip3 install torch torchvision torchaudio --index-url` https://download.pytorch.org/whl/cu118
2、安裝 ultralytics
?安裝ultralytics
運行:
conda create -n yolov8 python==3.8 -y
conda activate yolov8
pip install ultralytics==8.0