微信公众号:OpenCV学堂
关注获取更多计算机视觉与深度学习知识
引言
YOLOv8安装
YOLOv8代码库地址:
https://github.com/ultralytics/ultralytics
git clone https://github.com/ultralytics/ultralytics
pip install ultralytics
import ultralytics
ultralytics.checks()
运行截图如下:
YOLOv8系列模型速度、精度、参数统计表
01
推理测试
使用命令行
yolo predict model=yolov8n.pt source=zidane.jpg
运行截图如下:
02
导出ONNX格式
使用命令行
yolo mode=export model=yolov8n.pt format=onnx opset=11
运行截图如下:
自定义数据集训练
01
数据集制作与格式
这个跟YOLOv5完全一致,请参考我之前的文章
YOLOv5新版本6.x 自定义对象检测-从训练到部署
推荐学习视频课程:
https://ke.qq.com/course/5829598
记住一定要在
D:\python\my_yolov8_train_demo
D:\python\my_yolov8_train_demo\datasets
这个目录下就可以用了
训练执行命令行:
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
https://gist.github.com/cobryan05/7d1fe28dd370e110a372c4d268dcb2e5
python fixNvPe.py --input C:\Users\Administrator\AppData\Local\Programs\Python\Python38\Lib\site-packages\torch\lib\*.dll
总结
扫码查看OpenCV+OpenVIO+Pytorch系统化学习路线图
推荐阅读
CV全栈开发者说 - 从传统算法到深度学习怎么修炼
2022入坑深度学习,我选择Pytorch框架!
Pytorch轻松实现经典视觉任务
教程推荐 | Pytorch框架CV开发-从入门到实战
OpenCV4 C++学习 必备基础语法知识三
OpenCV4 C++学习 必备基础语法知识二
OpenCV4.5.4 人脸检测+五点landmark新功能测试
OpenCV4.5.4人脸识别详解与代码演示
OpenCV二值图象分析之Blob分析找圆
OpenCV4.5.x DNN + YOLOv5 C++推理
OpenCV4.5.4 直接支持YOLOv5 6.1版本模型推理
OpenVINO2021.4+YOLOX目标检测模型部署测试
比YOLOv5还厉害的YOLOX来了,官方支持OpenVINO推理