YOLOv5训练自己的数据集

上次说了如何将yolov5部署到自己的电脑上，并运行官方提供的案例

这次是如何用yolov5训练自己的数据集

1.首先准备好自己需要的图片，最好转为jpg格式，用labelImage来进行标注

labelImage可以在git上下载python代码 然后使用命令行启动，也可以直接去下载编译好的exe文件，双击直接启动
标注的时候需要注意

点击这里切换成yolo的格式 这样标注出来的数据就是txt格式的
不然选择voc那个就是xml格式的，yolov5训练的时候还不支持xml格式的，需要用python转成txt格式的
然后就可以开始绘制你的检测框了

这个路径下面的txt文件里面定义了你需要标注哪些类别

2.标注完毕之后，需要将现有的数据集划分为训练集，验证集等

执行下面代码 将数据集分开，得到的结果是几个txt文件，里面写了那些文件名是训练集，哪些文件名是测试集