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的简单训练教程啦,希望大家都可以成功。

猜你喜欢

转载自blog.csdn.net/weixin_45185577/article/details/129115488