使用Yolov3训练自己制作数据集,快速上手

在目标检测和分类这方面,Yolo可以快速很好的解决许多问题,这里总结了快速上手Yolov3的方法,直接快速训练自己的数据集使用。

我提供一个我自己已经调试通的源码包,包含了数据集和源代码,学习者可以先下载后配套着进行学习,我接下来的讲解,都将基于此源码包讲解,源码包下载链接地址为:添加链接描述提取码为:6vxv
下载好解压后的文件样纸见下:
使用Yolov3训练自己制作数据集,快速上手_第1张图片
下面开始介绍怎样使用源码包训练自己的数据集:

1.1 图片数据集和标签的存放位置,见下:
使用Yolov3训练自己制作数据集,快速上手_第2张图片
使用Yolov3训练自己制作数据集,快速上手_第3张图片
使用Yolov3训练自己制作数据集,快速上手_第4张图片
1.2 其中JPEGImages文件中的样纸见下:
使用Yolov3训练自己制作数据集,快速上手_第5张图片
1.3 其中Annotations文件中的样纸见下:
使用Yolov3训练自己制作数据集,快速上手_第6张图片
1.3.1 每个.xml文件中的内容见下:
使用Yolov3训练自己制作数据集,快速上手_第7张图片
在训练自己数据集的时候,只需要将自己的数据集图片拷贝到文件夹JPEGImages中,标签文件拷贝到文件Annotations中就行,不需要自己重新命名文件夹,直接用我给的框架就可以。

2 制作数据集标签:
关于制作VOC数据集yolo数据集的详细方法,可以参考我另外一篇博客,链接:添加链接描述
制作COCO数据集的详细方法见我另外一篇博客,链接:添加链接描述

3.1 在文件夹model_data文件中cls_classes.txt文件中写入打标签时的类名,见下:
使用Yolov3训练自己制作数据集,快速上手_第8张图片
3.2 文件夹model_data文件中yolo_anchors.txt文件,这里主要介绍一下文件中的内容,学习者不用修改,保持原有的默认即可,见下:
使用Yolov3训练自己制作数据集,快速上手_第9张图片
3.3 修改voc_annotion.py文件中classes_path的路径:
使用Yolov3训练自己制作数据集,快速上手_第10张图片
3.4 运行voc_anntion.py文件会生成6个训练要用到的.txt文件,6个.txt文件分别见下:
使用Yolov3训练自己制作数据集,快速上手_第11张图片

3.5 修改训练文件train.py中的classes_path,见下:
使用Yolov3训练自己制作数据集,快速上手_第12张图片
3.6 直接运行train.py文件就可以开始训练了,见下:
使用Yolov3训练自己制作数据集,快速上手_第13张图片

4.1 训练好模型后进行测试,将训练好的模型复制到yolo.py文件下,并修改classes_path,见下:
使用Yolov3训练自己制作数据集,快速上手_第14张图片
4.2 开始验证训练后模型的检测效果,直接运行文件predict.py文件,见下:
使用Yolov3训练自己制作数据集,快速上手_第15张图片
4.3.1 运行后的输出见下:
使用Yolov3训练自己制作数据集,快速上手_第16张图片
4.3.2 检测结果见下:
使用Yolov3训练自己制作数据集,快速上手_第17张图片
4.4.1 想用视频检测时的代码修改见下:
使用Yolov3训练自己制作数据集,快速上手_第18张图片
4.4.2 视频实时的检测效果见下(这里只是截取了其中一帧,运行代码视频是可以实时高效检测到人脸的):
使用Yolov3训练自己制作数据集,快速上手_第19张图片
以上就是使用Yolov3训练自己制作的数据集,快速上手的方法,学习者在使用的时候只需要按照我上面的步骤,修改几个文件参数就可以训练自己的数据集了,希望对正在学习Yolov3的你有所帮助,想快速上手学习Yolov5的学者,详见我另外一篇博客,多多支持,谢谢!

你可能感兴趣的:(python,Python,OpenCv,目标检测,深度学习,python,分类,功能测试)