引言
YOLOv8已經發布,文本是第一篇全面測試從訓練到部署的文章,詳細介紹每個細節。
YOLOv8安裝
如果你只是想使用而不是開發,強烈推薦通過pip安裝方式獲取YOLOv8包!YOLOv8安裝命令行
pip install ultralytics測試查詢版本號
import ultralytics ultralytics.checks()
運行截圖如下:
YOLOv8系列模型速度、精度、參數統計表
01
推理測試
使用命令行
yolo predict model=yolov8n.pt source=zidane.jpg
運行截圖如下:
注意:D:pythonmy_yolov8_train_demo為作者自建目錄。 第一次運行會自動下載yolov8n.pt模型到該目錄下。
02
導出ONNX格式
使用命令行
yolo mode=export model=yolov8n.pt format=onnx opset=11
運行截圖如下:
自定義數據集訓練
01
數據集制作與格式
這個跟YOLOv5完全一致,請參考我之前的文章
YOLOv5新版本6.x 自定義對象檢測-從訓練到部署
記住一定要在
D:pythonmy_yolov8_train_demo目錄下先創建一個datasets文件夾然后再把之前YOLOv5的自定義數據集給copy到
D:pythonmy_yolov8_train_demodatasets
這個目錄下就可以用了
訓練執行命令行:
yolo task=detect mode=train model=yolov8n.pt epochs=3 batch=1 data=datasets/dm_training/dataset.yaml
就可以直接開啟訓練了,截圖如下:
問題與解析
1. OSError: [WinError 1455] 頁面文件太小
執行命令行:
python -m pip install pefile
保存為fixNvPe.py, 運行命令行:
python fixNvPe.py --input C:UsersAdministratorAppDataLocalProgramsPythonPython38Libsite-packagestorchlib*.dll
即可解決。
2.從PyCharm IDE運行嵌入死循環 從命令行運行才是正確選擇
3.導出的ONNX格式模型無法加載 加上參數opset=11, 必須的!
總結
YOLOv8的數據集制作完全跟YOLOv5一致,模型部署推理跟YOLOv5及其相似,如果熟悉掌握YOLOv5的訓練與部署,看到此文就可以自動會了YOLOv8的訓練與推理了。
審核編輯:劉清
-
python
+關注
關注
56文章
4825瀏覽量
86377
原文標題:YOLOv8自定義數據集訓練到模型部署推理
文章出處:【微信號:CVSCHOOL,微信公眾號:OpenCV學堂】歡迎添加關注!文章轉載請注明出處。
發布評論請先 登錄
請問如何在imx8mplus上部署和運行YOLOv5訓練的模型?
怎樣使用PyTorch Hub去加載YOLOv5模型
【愛芯派 Pro 開發板試用體驗】愛芯元智AX650N部署yolov8s 自定義模型
TensorRT 8.6 C++開發環境配置與YOLOv8實例分割推理演示

教你如何用兩行代碼搞定YOLOv8各種模型推理

三種主流模型部署框架YOLOv8推理演示
基于OpenCV DNN實現YOLOv8的模型部署與推理演示

評論