1. 官网下载源码
2,官网下载预训练权重
并将预训练权重放在项目文件夹下。
3,安装所需依赖
python版本需要>=3.7才可以
pip install ultralytics
进入项目目录,打开命令台,pip安装ultralytics库。依赖环境只需要安装ultralytics库即可,yolov8已经将所需要的依赖集成到了ultralytics库。
4,预测一张图片
图片所在位置
注意,这里可以自己写一个简单的py文件进行预测。
预测的结果
5.训练自己的数据集
填写yaml文件
数据的结构如图所示
训练函数自己写一个简单的
6.注意事项
关闭wandb
分类数根据需要调整yolov8.yaml文件。需要用yolov8的n,s,m,l,x。即从yaml创建模型时添加对应的字母即可。