darknet 如何训练自己的数据集

本文档提供了一段MATLAB代码,用于根据XML注释文件生成VOC2007风格的数据集,包括trainval.txt、train.txt、test.txt和val.txt。通过调整代码中的比例参数,可以灵活地分配数据集用于训练、验证和测试。适用于使用darknet进行目标检测任务。
摘要由CSDN通过智能技术生成
darknet 官网:https://pjreddie.com/

1、使用matlab2017的 trainingImageLabeler命令对图像进行标记,并产生table数据类型;
2、将table中数据转化为指定的xml文件(img_to_xml_table.m);
3、将Annotations中的数据生成trainval.txt、train.txt、val.txt、test.txt文件(generate_label.m);
4、在darknet/scripts/下新建VOCdeckit文件夹,接着在VOCdeckit文件夹下新建VOC2007子文件夹。然后在VOC2007下新建Annotations、ImageSets、JPEGImages文件夹;
5、将xml文件放入Annotations文件夹下,在图片放入JPEGImages文件夹下,在ImageSets下新建Main文件,将trainval.txt、train.txt、val.txt、test.txt放入其中;
6、编辑voc_label.py,将sets修改为[('2007', 'train'), ('2007', 'val'), ('2007', 'test')],将classes修改为类别种类信息。另外,将os.system()中的2012_train.txt 2012_val.txt删除,并运行python voc_label.py命令;
7、编辑darknet/cfg/voc.data,将classes修改为类别数目,train和valid分别为2007_train.txt和2007_val.txt的路径;
8、编辑darknet/data/voc.names,填入类别的名称;
9、编辑darknet/data/yolov3-voc.cfg,修改类别信息;
10、运行./darknet detector train cfg/voc.data cfg/yolov3-voc.cfg darknet53.conv.74命令,即可训练自己的数据集。


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值