[yolov5] yolo的数据标签格式

yolov5 的标签格式

参考链接:

https://github.com/ultralytics/yolov5/issues/9816

翻译内容

你好!。感谢您询问YOLOv5🚀数据集格式。用于分割的XY坐标与用于长方体中心的标准坐标相同。

为了正确训练,您的数据必须为YOLOv5格式。有关数据集设置的完整文档以及开始培训您的第一个模型所需的所有步骤,请参阅我们的训练定制数据教程。以下是该教程的几个节选:

1.1 创建 dataset.yaml

COCO128是一个示例性的小教程数据集,由COCO train2017中的前128张图像组成。这些相同的128张图像用于训练和验证,以验证我们的训练管道是否会过拟合。data/COCO128.yaml是数据集配置文件(如下面所示),它定义了
1)数据集的根目录路径,train/val/test图像目录(或带有图像路径的*.txt文件)的相对路径

2)类名字典。

data/COCO128.yaml:

# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
path: ../datasets/coco128  # dataset root dir
train: images/train2017  # train images (relative to 'path') 128 images
val: images/train2017  # val images (relative to 'path') 128 images
test:  # test images (optional)

# Classes (80 COCO classes)
names:
  0: person
  1: bicycle
  2: car
  ...
  77: teddy bear
  78: hair drier
  79: toothbrush

1.2 创建 lables

使用Roboflow Annotate这样的工具为图像添加标签后,将标签导出为YOLO格式,并使用一个*.txt文件(如果图像中没有对象,则不需要*.txt文件)。*.txt文件规格如下:

  • 每个对象一行
  • 每行的格式是:class x_center y_center width height
  • 方框的坐标必须要归一化到(0-1),如果方框以像素为单位,请将x_center和width除以图像宽度,将y_center和height除以图像高度。
  • 类别必须从0开始,。

对应于下面图像的标签文件包含2个人(类别0)和一个领带(类别27):

在这里插入图片描述
在这里插入图片描述

1.3 组织目录

根据以下示例组织您的训练和val图像和标签。YOLOv5假设/coco128位于/YOLOv5目录旁边的/dataset目录中。YOLOv5通过将每个图像路径中/images/的最后一个实例替换为/labels/,自动查找每个图像的标签。例如:

../datasets/coco128/images/im0.jpg  # image
../datasets/coco128/labels/im0.txt  # label

Good luck 🍀 and let us know if you have any other questions!

  • 11
    点赞
  • 38
    收藏
    觉得还不错? 一键收藏
  • 5
    评论
您好!关于使用YOLOv5训练YOLO数据集,您可以按照以下步骤进行操作: 1. 准备数据集:首先,您需要准备一个包含标记边界框的数据集。确保每个图像都有对应的标签文件,该文件包含对象的类别和位置信息。 2. 数据预处理:使用YOLOv5提供的脚本对数据集进行预处理,以便符合模型的输入要求。您可以使用`python -m yolov5.tools.datasets`命令来完成这一步骤。 3. 划分数据集:将数据集划分为训练集和验证集,一般采用80%的数据作为训练集,20%的数据作为验证集。可以使用工具如`train_test_split`来实现。 4. 配置模型:根据您的需求,选择一个适合的YOLOv5模型。YOLOv5提供了多个预训练模型,您可以选择其一个作为起点,然后根据您的数据集进行微调。 5. 训练模型:使用训练数据集对模型进行训练。您可以使用`python -m yolov5.train`命令来启动训练过程。在训练过程,您可以调整超参数、学习率、迭代次数等来优化模型性能。 6. 评估模型:在训练过程结束后,使用验证集对模型进行评估,以了解模型在未见过的数据上的性能表现。YOLOv5提供了评估工具,可以使用`python -m yolov5.test`命令进行评估。 7. 模型推理:训练完成后,您可以使用YOLOv5模型进行目标检测任务。可以使用`python -m yolov5.detect`命令对图像或视频进行目标检测。 希望以上步骤对您有帮助!如果有更多问题,请随时提问。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值