精易论坛

标题: 基于Darknet框架的yolov4-tiny物体检测 [打印本页]

作者: fq19851220 时间: 2021-5-15 19:37
标题: 基于Darknet框架的yolov4-tiny物体检测
本帖最后由 fq19851220 于 2021-5-15 21:28 编辑

0. 上次发了个yolo检测调用支持库(地址在这),有些人想知道怎么训练,所以出个教程.
本教程演示竞技游戏dota2英雄检测的训练过程,仅为演示训练过程所以只检测一个英雄
拉比克!

1. 环境准备.
我的软件环境win10 cuda10.0 cudnn 7.6.5
显卡 1050ti
cuda下载地址  Cuda
cudnn下载地址cudnn
详细配置方法请百度,在此不展开.

2.数据准备. 找一个dota2直播间,准备了90张截图,剪出几张不参与训练后面作测试用

打开标记工具进行数据标注
命令行 yolo_mark.exe dota2/dota2 dota2/dota2_train.txt dota2/dota2.names
说明:
dota2/dota2--->采集的图片路径
dota2/dota2_train.txt--->图片相对于标注工具的相对路径文本集合,文本格式为:

dota2/dota2/2021-05-14_232036.png dota2/dota2/2021-05-14_232047.png
dota2/dota2/2021-05-14_232057.png
...

dota2/dota2.names--->类别名称因为只有一个英雄所以只有一个 0

开始标注

由于只有一个类,所以直接框选拉比克就行了,如果有多个类别,标注完一个英雄后,把上面第二行滑块条
往后拉继续标注就行.
...
  7分钟后标完成.
3.开始训练.
首先修改网络,复制一份yolov4-tiny.cfg 修改为yolov4-tiny-dota2.cfg
记事本打开yolov4-tiny-dota2.cfg
修改所有yolo层上面的convolutional中的filters后面的数字
数字改为 (类数量+5)*3  我们只有一个类所以为 (1+5)*3=18
filters=18
所有classes=改成类数量我们只有一个所以是classes=1
训练命令行 darknet.exe detector train dota2/dota2.data ./cfg/yolov4-tiny-dota2.cfg  -dont_show
说明:
dota2/dota2.data--->训练各种配置路径,文本格式为:
(classes = 1 --->类数量
train  = dota2/dota2_train.txt --->参与训练的图片路径
valid  = dota2/dota2_val.txt --->参与验证的图片路径
names = dota2/dota2.names --->类别名称因为只有一个英雄所以只有一个 0
backup = backupdota2/ --->训练模型的保存路径)

./cfg/yolov4-tiny-dota2.cfg--->我们上面修改的配置文件路径

-dont_show--->不显示窗口.
运行命令行,出现错误

显存不够,修改cfg文件
batch=64
subdivisions=8
batch改小或subdivisions改大
如果显存够大没有出错则不用修改.

....半小时1000迭代后训练完成,手动结束

损失为0.13,模型已保存在backupdota2
4.测试模型
使用上面的支持库测试一下效果,使用前面保存的未参与训练的图片测试

正确检出

未检出

正确检出

训练完成.

5.本教程仅演示框架的训练方法,数据较少,效果一般,训练需要英伟达显卡,建议4G显存以上太小的恐怕训练不了
本教程编写比较仓促,难免有错误和疏忽之处.

训练包下载地址

交流群:479282555

作者: 0010101 时间: 2021-5-15 19:53
坐等后续

作者: mz135135 时间: 2021-5-15 23:00
谢谢分享

作者: 胆小的老虎 时间: 2021-5-15 23:09
很棒！！！！！！！

作者: Khnu 时间: 2021-5-15 23:19
看看

作者: 秒速五厘米0127 时间: 2021-5-17 02:39
感谢分享，很给力！~

作者: 74596877 时间: 2021-5-17 03:01
支持一个!!!

作者: YuSiuLow 时间: 2021-5-17 03:11

作者: sanggeun 时间: 2021-5-17 10:40
路过~~~~~~~~~

作者: fengwubuzai 时间: 2021-5-17 22:26
看看是啥好东西

作者: ovo555 时间: 2021-5-18 10:13

作者: Lscaaaaa 时间: 2021-5-18 11:00
看看是啥好东西

作者: 红桃じ简简箪箪 时间: 2021-5-19 12:36
支持一个!!! 那个支持库支持y5么

作者: biody 时间: 2021-5-19 14:09
看看怎么操作的！

作者: ╰仅冇旳執著 时间: 2021-5-19 14:20
太秀了啊

作者: heyi8831 时间: 2021-5-19 17:28
66666666666666666666

作者: jjhcctv 时间: 2021-5-19 20:13

太秀了啊

作者: beven8627 时间: 2021-5-21 08:25
谢谢分享

作者: 17377771228 时间: 2021-5-22 17:08
学习学习学习学习学习学习学习学习学习学习学习学习学习学习学习学习学习学习学习学习学习学习学习学习

作者: 薛定谔的喵喵 时间: 2021-5-22 17:49

学习学习学习学习学习学习学习学习学习学习学习学习学习学习学习学习学习学习

作者: lq850815 时间: 2021-6-21 22:05
学习学习学习

作者: lq850815 时间: 2021-6-21 22:06
学习学习学习

作者: 莫尔多瓦 时间: 2021-8-14 15:43
百d连接失效了

作者: ht5124251 时间: 2021-9-9 23:22
A卡能玩吗

作者: leiwawa001 时间: 2021-12-2 09:04
谢谢分享。

作者: leonqp1030 时间: 2021-12-14 17:45
厉害啊！！！学习了！

作者: 乱剁 时间: 2022-2-27 12:37

学习学习学习

作者: twoseven271 时间: 2022-3-1 08:36
学习学习学习........

作者: 你好孙悟空 时间: 2022-3-19 12:05
环境配置不好

作者: 你好孙悟空 时间: 2022-3-21 15:29
提示缺少 opencv_world440.dll

作者: 洞子dark 时间: 2022-3-22 23:29

牛逼.谢谢楼主咯

作者: qepwq1314 时间: 2022-3-30 15:48

作者: a1055257427 时间: 2022-4-11 18:34
谢谢楼主很好的分享

作者: z188 时间: 2022-4-27 10:30
666666666666666666666666666666666

作者: 酷酷 时间: 2022-5-2 20:11
大哥有GPU版本的吗?CPU好慢

作者: 515667395 时间: 2022-8-24 22:42
多谢分享

作者: sskj2014 时间: 2022-10-14 14:41
版主加群验证通过下呗（☆真Dê很じove你）

作者: WSW120 时间: 2023-3-21 10:00
谢谢分享

作者: 百熙 时间: 2023-10-15 04:02
谢谢楼主的分享,不会整静态就出错....

作者: wyd168 时间: 2024-11-27 14:46
找不到这个群了479282555

作者: gn0129 时间: 2025-2-11 02:38
谢谢分享

欢迎光临精易论坛 (https://125.confly.eu.org/)