yolov8训练自己的数据集

1. 官网下载源码
yolov8训练自己的数据集_第1张图片

2,官网下载预训练权重
yolov8训练自己的数据集_第2张图片
并将预训练权重放在项目文件夹下。
yolov8训练自己的数据集_第3张图片

3,安装所需依赖
python版本需要>=3.7才可以

 pip install ultralytics

进入项目目录,打开命令台,pip安装ultralytics库。依赖环境只需要安装ultralytics库即可,yolov8已经将所需要的依赖集成到了ultralytics库。

在这里插入图片描述
4,预测一张图片
图片所在位置
yolov8训练自己的数据集_第4张图片
注意,这里可以自己写一个简单的py文件进行预测。
yolov8训练自己的数据集_第5张图片
预测的结果
yolov8训练自己的数据集_第6张图片
5.训练自己的数据集
填写yaml文件
数据的结构如图所示
yolov8训练自己的数据集_第7张图片
训练函数自己写一个简单的
yolov8训练自己的数据集_第8张图片
yolov8训练自己的数据集_第9张图片

6.注意事项
关闭wandb
yolov8训练自己的数据集_第10张图片
分类数根据需要调整yolov8.yaml文件。需要用yolov8的n,s,m,l,x。即从yaml创建模型时添加对应的字母即可。
在这里插入图片描述
yolov8训练自己的数据集_第11张图片

你可能感兴趣的:(深度学习,YOLO,python,开发语言)