码迷,mamicode.com
首页 > 其他好文 > 详细

yoloV3一步步训练自己的数据

时间:2018-09-25 01:11:02      阅读:4739      评论:0      收藏:0      [点我收藏+]

标签:nbsp   1.0   分享   link   http   读取   检测   dir   ros   

YOLOV3的主页:

https://pjreddie.com/darknet/yolo/

运行主页上的代码得到:

技术分享图片

 

首先使用一个开源的神经网络框架Darknet,使用C和CUDA,有CPU和GPU两种模式。

技术分享图片

1、下载darknet

编辑Makefile文件:

技术分享图片

2、编译:

make

如果遇到缺失包报错,将缺失的包装上去以后重新编译,重复下列两个操作,直到所有包都装好

make clean

make

安装好DarkNet之后,在darknet的子目录cfg/下已经有了一些网络模型的配置文件,在使用之前,需要下载好预训练好的权重文件yolo.weights(1.0 GB).

如果运行paper中的代码,则下载yolov3.weights后,运行以下命令就可以检测:

./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg

或者这个命令
./darknet detector test cfg/coco.data cfg/yolov3.cfg yolov3.weights data/dog.jpg
如果想检测多张图片,需要预先载入预训练的模型:
技术分享图片

3、数据制作

制作VOC数据集:

技术分享图片

根据上面的链接下载数据,得到

技术分享图片

具体的每一个值的计算方式是这样的:假设一个标注的boundingbox的左下角和右上角坐标分别为(x1,y1)(x2,y2),图像的宽和高分别为w,h

技术分享图片

归一化的中心点x坐标计算公式:((x2+x1) / 2.0)/ w

归一化的中心点y坐标计算公式:((y2+y1) / 2.0)/ h

归一化的目标框宽度的计算公式: (x2-x1) / w

归一化的目标框高度计算公式:((y2-y1)/ h

如果图片的高度、宽度未知,可以通过cv2读取。

imginfo = cv2.imread(img_dir + img_name + ‘.jpg‘).shape

imginfo里存的是图像的[h, w,通道数] ,这一需要注意imginfo结果里的顺序

 可以使用voc_label.py自动获取标签集合。

技术分享图片

总共20个类,运行以上脚本,得到了训练集的路径名称和对应的标签。

这个脚本主要是将训练集的路径写到test.txt中,并将xml中的label设置成训练需要的格式。

构建训练集:

cat 2007_train.txt 2007_val.txt 2012_*.txt > train.txt

构建好数据集,修改cfg/voc.data
技术分享图片

train指定训练集的路径,valid指定测试集的路径,names的文件中指定了该数据集标注的检测的类别,内容为:

技术分享图片

下载darknet53作为预训练的模型:

wget https://pjreddie.com/media/files/darknet53.conv.74
训练模型:
./darknet detector train cfg/voc.data cfg/yolov3-voc.cfg darknet53.conv.74
如果想使用多GPU,则使用以下的命令:
./darknet detector train cfg/voc.data cfg/yolov3-voc.cfg darknet53.conv.74 -gpus 0,1,2,3
如果想暂停训练,并且从断点开始训练则:
./darknet detector train cfg/coco.data cfg/yolov3.cfg backup/yolov3.backup -gpus 0,1,2,3
技术分享图片
其中 cfg/yolov3-voc.cfg设置了网络的结构
总结——使用自己的数据训练yolo模型:
1、下载数据集制作数据训练集的格式train.txt(或者自己写下载数据集的脚本)
技术分享图片
测试数据集:
技术分享图片

 标签数据集:

   技术分享图片

   技术分享图片

2、设置cfg/xxx.data,检测的类别数,训练集路径,测试集路径,检测类别

3、修改cfg模型设置:

   技术分享图片

 

3、下载预训练的模型,并且训练

 ./darknet detector train cfg/voc.data cfg/yolov3-voc.cfg darknet53.conv.74

 

yoloV3一步步训练自己的数据

标签:nbsp   1.0   分享   link   http   读取   检测   dir   ros   

原文地址:https://www.cnblogs.com/tinrry/p/9697055.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!