Darknet下的Yolo v3

Darknet下的Yolo v3 热爱秃头的每一天 2023-07-17 17:22:59 604

一、网址:https://github.com/AlexeyAB/darknet

二、训练自己的数据(检测人头):

1、经过1周训练的效果图,总体上效果还是很OK的,检测精度也比较高:

2、数据准备

2.1 标注工具

标注工具在我的其他博客里有说明,这部分跳过,标注的人头的时候注意选择成person即可,方便以后生成训练时使用的.txt文件。

2.2 标注人头

一般规则 - 您的训练数据集应包含您要检测的一组相对大小的对象(样本要充分)

  • train_network_width * train_obj_width / train_image_width ~= detection_network_width * detection_obj_width / detection_image_width
  • train_network_height * train_obj_height / train_image_height ~= detection_network_height * detection_obj_height / detection_image_heigh

2.3 训练数据转换:

需要把标注生成的.xml文件转换成.txt文件,切记名字要对应上:

2.4 生成train.txt和val.txt,比例大约为8:1

2.5 训练数据

1)修改Makefile配置,使用GPU训练

  1. GPU=1 #如果使用GPU设置为1,CPU设置为0
  2. CUDNN=1 #如果使用CUDNN设置为1,否则为0
  3. OPENCV=1 #如果调用摄像头,还需要设置OPENCV为1,否则为0
  4. OPENMP=0 #如果使用OPENMP设置为1,否则为0
  5. DEBUG=0 #如果使用DEBUG设置为1,否则为0
  6. CC=gcc
  7. NVCC=/home/user/cuda-9.0/bin/nvcc #NVCC=nvcc 修改为自己的路径
  8. AR=ar
  9. ARFLAGS=rcs
  10. OPTS=-Ofast
  11. LDFLAGS= -lm -pthread
  12. COMMON= -Iinclude/ -Isrc/
  13. CFLAGS=-Wall -Wno-unused-result -Wno-unknown-pragmas -Wfatal-errors -fPIC
  14. ...
  15. ifeq ($(GPU), 1)
  16. COMMON+= -DGPU -I/home/hebao/cuda-9.0/include/ #修改为自己的路径
  17. CFLAGS+= -DGPU
  18. LDFLAGS+= -L/home/hebao/cuda-9.0/lib64 -lcuda -lcudart -lcublas -lcurand #修改为自己的路径
  19. endif

编译程序:

make -j8

2)开始训练:

./darknet detector train cfg/head.data cfg/yolov3-head.cfg darknet53.conv.74 -gpus 0,1

3) 检测目标:

./darknet detect cfg/yolov3-head.cfg weights/yolov3_150000.weights data/dog.jpg

注:

该博客主要做记录使用,由于数据属于公司,不能提供给大家,有些问题还没有解决,希望专业人士能够指点一下,QQ:187100248.

问题如下:

1、训练到10w多次,结果不收敛,一直震荡,不知道如何解决:

2、小目标会检测不到,漏掉的较多;

3、检测时间如何优化的更好。

算法demo,封装了一个前期的演示效果,验证算法可以在windows上运行,需要的环境:

vs2015+cuda9.1+windows10+release版本即可:

demo网盘地址:链接:https://pan.baidu.com/s/10AW8SHdGarzsZKuIaTSvPA 密码:9al6

声明:本文内容由易百纳平台入驻作者撰写,文章观点仅代表作者本人,不代表易百纳立场。如有内容侵权或者其他问题,请联系本站进行删除。
红包 点赞 收藏 评论 打赏
评论
0个
内容存在敏感词
手气红包
    易百纳技术社区暂无数据
相关专栏
置顶时间设置
结束时间
删除原因
  • 广告/SPAM
  • 恶意灌水
  • 违规内容
  • 文不对题
  • 重复发帖
打赏作者
易百纳技术社区
热爱秃头的每一天
您的支持将鼓励我继续创作!
打赏金额:
¥1易百纳技术社区
¥5易百纳技术社区
¥10易百纳技术社区
¥50易百纳技术社区
¥100易百纳技术社区
支付方式:
微信支付
支付宝支付
易百纳技术社区微信支付
易百纳技术社区
打赏成功!

感谢您的打赏,如若您也想被打赏,可前往 发表专栏 哦~

举报反馈

举报类型

  • 内容涉黄/赌/毒
  • 内容侵权/抄袭
  • 政治相关
  • 涉嫌广告
  • 侮辱谩骂
  • 其他

详细说明

审核成功

发布时间设置
发布时间:
是否关联周任务-专栏模块

审核失败

失败原因
备注
拼手气红包 红包规则
祝福语
恭喜发财,大吉大利!
红包金额
红包最小金额不能低于5元
红包数量
红包数量范围10~50个
余额支付
当前余额:
可前往问答、专栏板块获取收益 去获取
取 消 确 定

小包子的红包

恭喜发财,大吉大利

已领取20/40,共1.6元 红包规则

    易百纳技术社区