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

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

我提供一个我自己已经调试通的源码包,包含了数据集和源代码,学习者可以先下载后配套着进行学习,我接下来的讲解,都将基于此源码包讲解,源码包下载链接地址为:添加链接描述提取码为:6vxv
下载好解压后的文件样纸见下:

使用Yolov3训练自己制作数据集,快速上手
下面开始介绍怎样使用源码包训练自己的数据集:

1.1 图片数据集和标签的存放位置,见下:

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

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

3.1 在文件夹model_data文件中cls_classes.txt文件中写入打标签时的类名,见下:

使用Yolov3训练自己制作数据集,快速上手
3.2 文件夹model_data文件中yolo_anchors.txt文件,这里主要介绍一下文件中的内容,学习者不用修改,保持原有的默认即可,见下:
使用Yolov3训练自己制作数据集,快速上手
3.3 修改voc_annotion.py文件中classes_path的路径:
使用Yolov3训练自己制作数据集,快速上手
3.4 运行voc_anntion.py文件会生成6个训练要用到的.txt文件,6个.txt文件分别见下:
使用Yolov3训练自己制作数据集,快速上手

3.5 修改训练文件train.py中的classes_path,见下:

使用Yolov3训练自己制作数据集,快速上手
3.6 直接运行train.py文件就可以开始训练了,见下:
使用Yolov3训练自己制作数据集,快速上手

4.1 训练好模型后进行测试,将训练好的模型复制到yolo.py文件下,并修改classes_path,见下:

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

Original: https://blog.csdn.net/qq_40280673/article/details/125162408
Author: 佐咖
Title: 使用Yolov3训练自己制作数据集,快速上手

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/661987/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球