YOLOV5实战教程（超级详细图文教程）！！！_yolov5训练

提取码：8r4i

二、使用coco128数据集进行训练

2.1 数据集准备

下好的数据集应该是如下图所示

images文件夹里存放的是我们要训练的图片，labels文件夹里存放的是打过标签的图片

目标检测需要手动进行打标签，这个我会在后面介绍使用自己数据集时进行详细介绍

2.2 进行训练

编译器这里我使用的Pycharm，打开项目如下图

这里我们需要知道两个yaml文件

第一个是data文件夹下的coco128.yaml     数据集参数文件

打开这个文件如下图，这个文件里是我们需要更改训练集以及测试集路径的地方，下面的names是每一个标签，可以看到coco128数据集里的标签数量是非常大的，我们平时自己玩的时候一般不会用到这么多标签。

接着我们需要了解的另一个文件是models文件夹下的从yolov5l到yolov5s的训练权重

我们打开yolov5s这个文件，如下图。这里nc是我们的标签数量。

以上两个文件我们目前都不用改，只是了解一下方便我们后面使用自己数据集进行测试时讲解

接着我们打开train.py文件，找到这个位置

–weights 初始训练权重

–cfg 模型参数位置

–data 数据集参数文件

直接开始运行train.py