划分数据集，训练自己的数据集。

数据集划分是跟着up主魔傀面具做的，很好用很方便，推荐给大家，顺便做个例子讲一下怎么使用

把自己的图片数据集放在dataset/VOCdevkit/JPEGImages里面，看看自己的数据集格式，是JPEG还是png格式的还是其他。

然后就是把标注的文件放在dataset/VOCdevkit/Annotations，如果你的格式是xml格式

如果是txt格式的话，放在dataset/VOCdevkit/txt

标注文件的格式看的方法

看我的文件后缀就是.xml

或者打开属性

图片格式的查看也类似

我用一些数据来做个例子，我的数据集是100张，格式是JPG，标注文件是xml格式的

点击右键，不管是用pycharm还是vscode都可以

我用vscode为例子

如果是xml格式的标注文件，先运行xml2txt.py文件

在这之前，先配好自己的环境

vscode配置anaconda环境-CSDN博客

设置自己图片的格式，我的是jpg

在vscode上方打开终端

先进入的dataset文件夹里

输入

cd dataset

最新一行的最后面是dataset，就说明进入dataset文件夹了

输入

python xml2txt.py

这就是转换好了，下面出现的的是类别的名字，把这个复制到data.yaml里面

接着是划分数据集，点开split_data.py

我的图片格式是jpg，自己看自己的格式是什么

这是划分的比例，一般来说都是8:1:1，这个就不用动了

如果想划分成别的也可以，val_size是验证集占的比例，test_size是测试集占的比例，看你想设置多少，也可以设置为0

接着就是在终端输入 

python split_data.py

回车就可以了 

这是我划分的结果，按照8:1:1分的，结果在dataset文件夹下又image和labels两个文件夹里放着，里面都是对应着放的

终端操作教程，前提是你的python文件都已经设置好了

按住shift加右键点击dataset文件

先选择自己的环境

这里不会的看我的其他文章，里面有介绍使用的

conda activate 自己环境的名字

如果自己的数据集格式是txt的直接运行

python split_data.py

自己的数据集是xml格式的，先运行

python xml2txt.py

 然后再运行

python split_data.py

 这时候我的标注文件已经转换过txt了，这是我的运行结果

最后附上去代码链接

链接：https://pan.baidu.com/s/1wBYbKemcK67nVxmtSPundQ?pwd=0pdc 
提取码：0pdc 
--来自百度网盘超级会员V2的分享