提取码：8r4i
二、使用coco128数据集进行训练
2.1 数据集准备
下好的数据集应该是如下图所示
目标检测需要手动进行打标签，这个我会在后面介绍使用自己数据集时进行详细介绍
2.2 进行训练
编译器这里我使用的Pycharm，打开项目如下图
这里我们需要知道两个yaml文件
第一个是data文件夹下的coco128.yaml 数据集参数文件
打开这个文件如下图，这个文件里是我们需要更改训练集以及测试集路径的地方，下面的names是每一个标签，可以看到coco128数据集里的标签数量是非常大的，我们平时自己玩的时候一般不会用到这么多标签。
接着我们需要了解的另一个文件是models文件夹下的从yolov5l到yolov5s的训练权重
我们打开yolov5s这个文件，如下图。这里nc是我们的标签数量。
以上两个文件我们目前都不用改，只是了解一下方便我们后面使用自己数据集进行测试时讲解
接着我们打开train.py文件，找到这个位置
–weights 初始训练权重
–cfg 模型参数位置
–data 数据集参数文件
直接开始运行train.py