YOLOv5笔记（二）标注自己的数据集

Lulin Lv2
  2022-07-25 10:54:34 2022-07-25 10:54:34 Created   2023-10-16 11:33 2023-10-16 11:33 Updated      235 Words  1 Mins  

YOLOv5笔记（二）标注自己的数据集

0. 自定义数据集

使用coco数据集格式，创建如下文件夹结构

1
2
3
4
5
6
7
- trainImages
    - train  # 训练集
        - images    # 图片
        - labels    # 标签
    - val   # 验证集
        - images    # 图片
        - labels    # 标签

图片按照7:3的比例分布到训练集和验证集，分别放到images目录下。

1. 安装labelImg图像标注工具

conda create -n label python=3.8

conda activate label

pip install labelImg

2. 添加自定义标签

修改文件 labelImg/data/predefined_classes.txt:

1
2
dapeng
piaofuwu

3. 图像标注

运行labelImg

labelImg

选择 查看，勾选 自动保存模式


点击左侧 打开目录，选择训练集或数据集中 images 目录，点击 改变存放目录，选择训练集或数据集中 labels 目录

点击如下图所示按钮，将数据集保存格式切换到yolo

标注快捷键：W 创建标注框，A 切换上一张图片，D 切换下一张图片

  • Title: YOLOv5笔记（二）标注自己的数据集
  • Author: Lulin
  • Created at : 2022-07-25 10:54:34
  • Updated at : 2023-10-16 11:33:00
  • Link: https://blog.lllin.top/2022/07/25/yolov5-02/
  • License: This work is licensed under CC BY-NC-SA 4.0.
 Comments
On this page
YOLOv5笔记（二）标注自己的数据集
  1. 0. 自定义数据集
  2. 1. 安装labelImg图像标注工具
  3. 2. 添加自定义标签
  4. 3. 图像标注