Darknet下的Yolo v3
一、网址: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训练
GPU=1 #如果使用GPU设置为1,CPU设置为0
CUDNN=1 #如果使用CUDNN设置为1,否则为0
OPENCV=1 #如果调用摄像头,还需要设置OPENCV为1,否则为0
OPENMP=0 #如果使用OPENMP设置为1,否则为0
DEBUG=0 #如果使用DEBUG设置为1,否则为0
CC=gcc
NVCC=/home/user/cuda-9.0/bin/nvcc #NVCC=nvcc 修改为自己的路径
AR=ar
ARFLAGS=rcs
OPTS=-Ofast
LDFLAGS= -lm -pthread
COMMON= -Iinclude/ -Isrc/
CFLAGS=-Wall -Wno-unused-result -Wno-unknown-pragmas -Wfatal-errors -fPIC
...
ifeq ($(GPU), 1)
COMMON+= -DGPU -I/home/hebao/cuda-9.0/include/ #修改为自己的路径
CFLAGS+= -DGPU
LDFLAGS+= -L/home/hebao/cuda-9.0/lib64 -lcuda -lcudart -lcublas -lcurand #修改为自己的路径
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
- 分享
- 举报
-
浏览量:2025次2023-01-12 15:47:48
-
浏览量:9227次2020-12-03 17:46:05
-
浏览量:1275次2024-02-20 10:27:52
-
2024-01-22 16:01:53
-
浏览量:624次2024-01-22 15:27:25
-
浏览量:811次2023-07-05 10:18:03
-
浏览量:763次2023-04-17 15:47:23
-
浏览量:5410次2021-08-09 16:11:19
-
浏览量:10531次2021-04-27 00:28:09
-
浏览量:1057次2023-12-18 17:48:09
-
浏览量:6682次2021-05-01 14:14:55
-
浏览量:6558次2021-06-14 02:59:16
-
浏览量:1175次2024-02-19 17:07:05
-
浏览量:5338次2021-05-02 18:00:46
-
浏览量:9833次2021-04-30 00:31:47
-
2024-01-06 11:14:41
-
浏览量:1626次2024-03-01 16:01:03
-
浏览量:1567次2024-02-04 17:13:47
-
浏览量:5951次2021-05-17 16:48:29
-
广告/SPAM
-
恶意灌水
-
违规内容
-
文不对题
-
重复发帖
热爱秃头的每一天
感谢您的打赏,如若您也想被打赏,可前往 发表专栏 哦~
举报类型
- 内容涉黄/赌/毒
- 内容侵权/抄袭
- 政治相关
- 涉嫌广告
- 侮辱谩骂
- 其他
详细说明