yolo模型训练所需的数据集的划分教程

数据集的划分是使用yolo网络自主训练智能检测模型的前提步骤,本次将使用以及标注好的数据进行数据划分.之后会教大家如何标注数据

1.数据文件夹的创建

1.1 在yolov5的总文件夹中,有一个data的文件夹,在此文件夹下分别创建如下图两个文件夹,名字可更改为其他,但不要出现中文

1.2 其中一个文件夹中放置所有".xml"后缀的文件,这种文件是以及标注过的图片的信息

1.3 另一个文件夹中放置所有的图片,注意图片的后缀类型是什么样的,如下图所示,我的图片都是png格式的

注意: 标注的文件夹内所有文件的名称和图片文件夹中所有的名称都是一一对应相同的,并且这两个文件夹中的数量也应该相同

2.python文件的更改

2.1 打开data.py文件,在文件开头更改如图所示三个内容依次是:

1. xml文件夹所在的目录
2. 图片文件夹所在的目录