yolov8训练自己的数据集

发布时间:2024-01-03 12:36:49

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创建模型时添加对应的字母即可。
在这里插入图片描述
在这里插入图片描述

文章来源:https://blog.csdn.net/qq_53817374/article/details/135358621
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。