yolov5训练自己的数据集

yolov5训练自己的数据集

1 准备自己的数据集

1 按以下方式放置图片与标签

2 设置数据集参数

在data文件下新建一个yaml文件，仿照coco.yaml.

train:训练集图片的路径

val：验证集图片的路径

nc：自己数据集的类别总数

names：数据集里的类别，顺序与图片的标签相对应。

2 修改模型的配置文件

以最小模型为例，在models文件下打开yolov5s.yaml。修改类别总数nc

3 修改Train.py文件，训练自己的数据集

打开train.py。使用预训练模型，将data更改为自己的数据集配置文件。

(img-3FSfdPVa-1625834665296)]

4 运行train.py.