darknet训练自己的数据

voc数据格式:

  • –VOC
    • –Annotations //存放xml文件,每一个xml对应一张图像及各个目标的位置和类别信息
    • –ImageSets
      • –Main //存放train.txt、val.txt等,该文本文件里为图像名字(无后缀无路径)
      • –Layout
      • –Segmentation
    • –JPEGImages //原始图像
    • –SegmentationClass
    • –SegmentationObject

数据准备:

  1. 根据voc数据格式新建所有文件夹;
  2. 将图片放到JPEGImages下;
  3. ImageSets/Main下,新建train.txt存放训练图片名称,val.txt存放验证图片名称;
  4. 用lableimage生成xml
  5. 修改darknet\scripts\voc_label.py中对应文件位置;
    • 注意Difficult首字母大写;
    • 运行完后会在VOC新生成文件夹labels,同时在scripts\下应该生成了2017_train.txt、2017_val.txt,txt文件存放图片路径
    • labels文件夹存放图片坐标及其对应的类别
      txt文件可放到JPEGImages ,也可不放,注意文件夹名称不同时,需要放。
      这里写图片描述

训练流程:

  1. 新建voc.data
    classes= 2
    train  = E:/darknet/build/darknet/x64/data/myvoc/2017_train.txt //训练文件
    valid  = E:/darknet/build/darknet/x64/data/myvoc/2017_val.txt  //验证文件
    names = E:/darknet/build/darknet/x64/data/myvoc/voc.names  //类别名称
    backup = E:/darknet/build/darknet/x64/data/myvoc/backup/  //存放训练完成的权重,100/200/1000次
  1. 新建voc.names
    • 存放类别
    person
    car
  1. 修改tiny-yolo-voc.cfg
    • 修改region/classes
    • 修改最后convolutional/filters,filter=num×(classes + coords + 1)=5×(2+4+1)=35
  2. 训练
    • darknet detector train E:\darknet\build\darknet\x64\data\myvoc\voc.data E:\darknet\build\darknet\x64\data\myvoc\tiny-yolo-voc.cfg
    • ./darknet detector train data/myvoc/voc.data data/myvoc/tiny-yolo-voc.cfg
  3. 加载训练好的权重进行训练
  4. 测试
    • darknet.exe detector test data/myvoc/voc.data data/myvoc/tiny-yolo-voc.cfg data/myvoc/tiny-yolo-voc_200.weights E:/darknet/scripts/VOCdevkit/myVOC2017/JPEGImages/s8.jpg -thresh 0.3
  • 0
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值