引言
YOLOv8已經(jīng)發(fā)布,文本是第一篇全面測(cè)試從訓(xùn)練到部署的文章,詳細(xì)介紹每個(gè)細(xì)節(jié)。
YOLOv8安裝
如果你只是想使用而不是開發(fā),強(qiáng)烈推薦通過(guò)pip安裝方式獲取YOLOv8包!YOLOv8安裝命令行
pip install ultralytics測(cè)試查詢版本號(hào)
import ultralytics ultralytics.checks()
運(yùn)行截圖如下:
YOLOv8系列模型速度、精度、參數(shù)統(tǒng)計(jì)表
01
推理測(cè)試
使用命令行
yolo predict model=yolov8n.pt source=zidane.jpg
運(yùn)行截圖如下:
注意:D:pythonmy_yolov8_train_demo為作者自建目錄。 第一次運(yùn)行會(huì)自動(dòng)下載yolov8n.pt模型到該目錄下。
02
導(dǎo)出ONNX格式
使用命令行
yolo mode=export model=yolov8n.pt format=onnx opset=11
運(yùn)行截圖如下:
自定義數(shù)據(jù)集訓(xùn)練
01
數(shù)據(jù)集制作與格式
這個(gè)跟YOLOv5完全一致,請(qǐng)參考我之前的文章
YOLOv5新版本6.x 自定義對(duì)象檢測(cè)-從訓(xùn)練到部署
記住一定要在
D:pythonmy_yolov8_train_demo目錄下先創(chuàng)建一個(gè)datasets文件夾然后再把之前YOLOv5的自定義數(shù)據(jù)集給copy到
D:pythonmy_yolov8_train_demodatasets
這個(gè)目錄下就可以用了
訓(xùn)練執(zhí)行命令行:
yolo task=detect mode=train model=yolov8n.pt epochs=3 batch=1 data=datasets/dm_training/dataset.yaml
就可以直接開啟訓(xùn)練了,截圖如下:
問(wèn)題與解析
1. OSError: [WinError 1455] 頁(yè)面文件太小
執(zhí)行命令行:
python -m pip install pefile
保存為fixNvPe.py, 運(yùn)行命令行:
python fixNvPe.py --input C:UsersAdministratorAppDataLocalProgramsPythonPython38Libsite-packagestorchlib*.dll
即可解決。
2.從PyCharm IDE運(yùn)行嵌入死循環(huán) 從命令行運(yùn)行才是正確選擇
3.導(dǎo)出的ONNX格式模型無(wú)法加載 加上參數(shù)opset=11, 必須的!
總結(jié)
YOLOv8的數(shù)據(jù)集制作完全跟YOLOv5一致,模型部署推理跟YOLOv5及其相似,如果熟悉掌握YOLOv5的訓(xùn)練與部署,看到此文就可以自動(dòng)會(huì)了YOLOv8的訓(xùn)練與推理了。
審核編輯:劉清
-
python
+關(guān)注
關(guān)注
56文章
4822瀏覽量
85874
原文標(biāo)題:YOLOv8自定義數(shù)據(jù)集訓(xùn)練到模型部署推理
文章出處:【微信號(hào):CVSCHOOL,微信公眾號(hào):OpenCV學(xué)堂】歡迎添加關(guān)注!文章轉(zhuǎn)載請(qǐng)注明出處。
發(fā)布評(píng)論請(qǐng)先 登錄
相關(guān)推薦
基于YOLOv8實(shí)現(xiàn)自定義姿態(tài)評(píng)估模型訓(xùn)練

請(qǐng)問(wèn)如何在imx8mplus上部署和運(yùn)行YOLOv5訓(xùn)練的模型?
怎樣使用PyTorch Hub去加載YOLOv5模型
【愛芯派 Pro 開發(fā)板試用體驗(yàn)】愛芯元智AX650N部署yolov8s 自定義模型
TensorRT 8.6 C++開發(fā)環(huán)境配置與YOLOv8實(shí)例分割推理演示

Pytorch Hub兩行代碼搞定YOLOv5推理

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

三種主流模型部署框架YOLOv8推理演示
YOLOv8實(shí)現(xiàn)任意目錄下命令行訓(xùn)練

基于YOLOv8的自定義醫(yī)學(xué)圖像分割

YOLOv8+PyQT5打造細(xì)胞計(jì)數(shù)與識(shí)別應(yīng)用說(shuō)明

基于OpenCV DNN實(shí)現(xiàn)YOLOv8的模型部署與推理演示

RV1126 yolov8訓(xùn)練部署教程

評(píng)論