yolov5-7.0简单训练教程

---

前言

yolov5-7.0简单训练教程。

---

一、下载yolov5-7.0代码

可以自己去git上下载，没有梯子的点这里去百度网盘下载资源

二、使用步骤

1.下载资源包

点击下载资源包：objectdetection_script-master

将下载到的脚本目录中的yolo文件夹内的文件复制到yolov5-7.0的根目录下。

Annotations：存放VOC格式的标注的XML文件
JPEGImages：存放数据集里的图片，图片后缀需要一致，比如都是jpg或者png等等，不支持混合的图片后缀格式，比如一些是jpg，一些是png。
txt：存放yolo格式的标注文件。（运行xml2txt.py）

2.运行xml2txt.py

运行xml2txt.py,在这个文件中其会把Annotations中的XML格式标注文件转换到txt中的yolo格式标注文件。其中xml2txt.py中的postfix参数是JPEGImages的图片后缀,修改成图片的后缀即可，默认为jpg。比如我的图片都是png后缀的，需要把postfix修改为png即可。
其中运行这个文件的时候，输出信息会输出你的数据集的类别，你需要把类别列表复制到data.yaml中的names中，并且修改nc为你的类别数，也就是names中类别个数。

3.运行split_data.py

这个文件是划分训练、验证、测试集。其中支持修改val_size验证集比例和test_size测试集比例，可以在split_data.py中找到对应的参数进行修改，然后postfix参数也是你的图片数据集后缀格式，默认为jpg，如果你的图片后缀不是jpg结尾的话，需要修改一下这个参数。

4.开始训练

点击根目录下的train.py，调整相应的配置，就可以开始训练了。

---

总结

以上就是yolov5-7.0的简单训练教程啦，希望大家都可以成功。