利用yolov5训练自己的数据集(适合小白第一次训练数据集)

yolov5是yolo系列比较经典的一个算法,因此想跑一跑yolov5的代码,这是我第一个运行的代码,什么都还不懂 ,在下载完yolov5的代码之后特别茫然,不知道从哪里下手。这篇文章记录一下。

1.yolov5下载

点击 YOLOv5官方地址
在这里插入图片描述
点击code,下载为zip文件,或者复制https链接,通过git下载(这个方法需要使用git,需要在自己电脑里安装git不是很方便,所以推荐第一种)
在这里插入图片描述
下载好后解压,导入到pycharm中。

2. pycharm环境配置

可以参考这篇pytorch的安装(包括anoconda和pytorch与torch.cuda.is_available()返回false的解决方案)

3.测试是否导入的项目可用

3.1安装必要的包

首先需要安装必要的包,所有需要的包都已经写在了requirements.txt中了。打开terminal,进入本项目的annoconda环境
在这里插入图片描述
使用命令安装全部包


pip install -r requirements.txt

在这里插入图片描述

4.修改代码

在train.py中有这样一大段代码,这是有关配置相关的
在这里插入图片描述

5.开始训练(数据集为自带的coco128)

5.1使用权重文件训练

这是最常用的方式,通过权重文件训练,可以直接运行train.py训练,但是由于github连接速度太慢的问题,可能会报错,因此也可手动下载权重文件。

5.2不使用权重文件训练

不使用权重文件序列不推荐,需要训练很久才会收敛。

6.使用自己的数据集训练

yolo的代码使用的自己的数据集格式,而常见的数据集格式为coco数据集格式,因此,如果使用的是coco数据集,需要将coco数据集格式转为yolo格式。可以参考这篇博文。
coco数据集转yolo数据集(简单易懂)

6.1修改代码

第一处:将个人的数据集放在和代码自带的数据集下的同一目录
在这里插入图片描述
第二处:创建yml文件
在data文件夹下创建一个yml文件,保存配置信息
在这里插入图片描述
yml中的需要写的字段
(特别注意:yml是配置文件,写过springboot比较熟悉,这里每个字段必须要顶格写,此外,数据要满足 字段冒号空格字段),不要把空格删除了)
在这里插入图片描述

修改train.py中的代码,图中红框需要改成刚刚配置的文件名
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/qq_43471945/article/details/128163455
今日推荐