yolo训练自己的数据集

本文介绍了如何使用darknet训练YOLO模型,从制作VOC格式的数据集,包括使用LabelImg进行标注,到数据集的划分,以及训练前的配置步骤。通过traindata.py和trans.py脚本,分别生成训练和验证所需的txt文件,并提供了相关资源链接。
摘要由CSDN通过智能技术生成

前言

使用darknet来配置yolo训练自己的数据集是很很方便的

制作自己的数据集

训练自己的数据集首先要制作好训练集,制作数据集,yolo使用的数据集是voc格式的.制作数据集之前先要把自己的数据(一般是图片)进行标注,标注使用的工具是LabelImg,安装及使用方法可参考这篇博客.

标注完成后还需要进行一些处理,按照voc格式创建一些文件夹和必要的文件,我参考的是这篇博客.

制作过程也简单讲一下:
a.在“Image”文件夹下存放所有的图片样本(包括训练数据和验证数据,而且最好是jpg格式)
b.下载labelImg(一种图像标记工具,给图像中的目标打上标签,且可以生成训练需要的xml格式),具体的使用方法可以百度,操作起来很简单。
c.与“Image”文件夹同级新建“xml”文件夹,“xml”文件夹存放labelImg得到的所有图片样本的标签。
d.现在就是要将所有的样本分成训练集和验证集,而且要将训练集和验证集对应的xml也分开。这里下载python脚本,直接放在“Image”和“xml”文件夹同级路径。

具体的:

运行traindata.py:

生成tr

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值