【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)

如何制作和训练自己的数据集

  • 看yolov5官网
  • 创建数据集
    • 1.搜索需要的图片
    • 2.创建标签
      • 标注数据集地址:
      • 放入图片后选择目标检测
      • 创建文档,每个标签写在单独的一行
      • 上传结果
      • 此处可以编辑类别
      • 把车框选选择类别即可
      • 导出数据
    • 3.新建一个目录放数据
      • 写yaml文件
    • 4. 测试训练效果
  • 防止pychram检索数据集的方法

看yolov5官网

【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第1张图片
首先是创建.yaml数据集
1)首先指明数据集的根目录在哪里,训练集,验证集,测试集他们文件夹的相对路径
2)指定训练中有多少个类
3)指定这些类代表什么含义
【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第2张图片

创建数据集

1.搜索需要的图片

这里名字最好保存为英文,保存在data下面
【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第3张图片

2.创建标签

标注数据集地址:

https://www.makesense.ai/

【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第4张图片

放入图片后选择目标检测

【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第5张图片

创建文档,每个标签写在单独的一行

【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第6张图片

【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第7张图片
【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第8张图片

上传结果

【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第9张图片
【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第10张图片

此处可以编辑类别

【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第11张图片
从上往下依次是
标签列表
导入图片
导入标注
导出标注
用训练好的检测

【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第12张图片
一般用矩阵标注

把车框选选择类别即可

【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第13张图片

导出数据


【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第14张图片
需要归一化

3.新建一个目录放数据

【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第15张图片

继续在该目录下新建
【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第16张图片
新建目录如图
【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第17张图片
把图片和标注的数据放入
【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第18张图片

写yaml文件

直接复制一个,并命名为mydata
【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第19张图片
【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第20张图片
更改如图
【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第21张图片
更改train.py里的数据集路径
在这里插入图片描述
开始训练
在这里插入图片描述

4. 测试训练效果

【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第22张图片
拷贝到 detect.py
【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第23张图片

【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第24张图片
更改后
【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第25张图片

点击运行
【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第26张图片

防止pychram检索数据集的方法

【 YOLOv5】目标检测 YOLOv5 开源代码项目调试与讲解实战(4)-自制数据集及训练(使用makesense标注数据集)_第27张图片

你可能感兴趣的:(YOLO,目标检测,目标跟踪,学习,笔记)