Yolov8中官方数据集格式转化DOTA->OBB函数的坑

前言:

        在使用Yolov8官方提供的DOTA数据集转OBB数据集过程中,发现我的新的标签文件根本没有生成,也就是convert_dota_to_yolo_obb没有产生效果,后续翻阅代码才发现这是一个小坑,下文将细细道来。

出现的问题:

         我们都知道要让Yolov8进行旋转框标注,需要对数据集进行格式转化,转为obb格式,官方之前提供了官方的函数,可以让DOTA格式的数据集一键转化为OBB格式,但是我运行之后没有反应,只生成了文件夹,但是新的转化好的标签文件没有出现。

官方转化代码如下:

from ultralytics.data.converter import convert_dota_to_yolo_obb
convert_dota_to_yolo_obb(r'你的数据集路径')
#关于dataobb文件下的目录下面会详细说明

解决方案:

         我们ctrl+鼠标左键点击convert_dota_to_yolo_obb函数,跳转到converter.py这个文件,直接拉到代码409-410行,你会发现有个大!大!大!大!的坑,他居然加了个不是png格式的图片就跳过的这个判断,真坑啊!

        解决方法就是注释掉就行,大功告成!

  • 12
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
### 回答1: YOLOv5_obb数据集是一种基于YOLO (You Only Look Once)算法改进的目标检测数据集,用于检测物体的外接矩形框(Oriented Bounding Box,OBB)。 传统的YOLO算法通常使用水平矩形框来表示物体的位置和尺寸,但对于某些具有倾斜或旋形状的物体,使用OBB可以更准确地描述其位置和方向。因此,YOLOv5_obb数据集使用OBB来定义物体的边界框,以提高目标检测的准确性。 YOLOv5_obb数据集通常包含大量带有OBB标签的图像样本,每个样本都包含一个或多个物体的OBB标注信息。这些OBB标注信息一般由矩形框的四个顶点的位置和物体类别标签组成。 使用YOLOv5_obb数据集可以训练一个可以检测和定位具有任意形状的物体的目标检测模型。通过学习样本图像物体的OBB信息,模型可以准确地识别和定位这些物体。 此外,YOLOv5_obb数据集还可用于解决一些特定应用场景的目标检测问题,例如航空影像、卫星图像或城市场景的目标检测。这些场景下的物体可能具有复杂的形状和旋角度,使用OBB可以更好地适应这些特殊要求。 总之,YOLOv5_obb数据集是一种为了解决特定物体形状和方向的目标检测问题而设计的数据集。通过使用OBB标注信息,可以提高目标检测模型在这些场景下的表现和准确性。 ### 回答2: YOLOv5是一种目标检测算法,而YOLOv5-OBB则是基于YOLOv5的一种针对有方向边界框目标的检测模型。OBB全称为Oriented Bounding Box,即有方向的边界框。 YOLOv5-OBB数据集是针对有方向边界框目标检测任务构建的数据集。在常规的边界框目标检测任务,边界框是水平的,即边界框的长宽与坐标轴平行。而在有方向边界框目标检测任务,边界框可以按照物体的朝向进行旋。 为了支持YOLOv5-OBB模型的训练和评估,需要一个包含有方向边界框的标注数据集。该数据集需要提供每个目标物体的类别标签、边界框的顶点坐标和角度信息。 标注一个有方向边界框通常需要指定边界框的心坐标、边界框的宽度和高度、以及边界框的旋角度。在训练过程YOLOv5-OBB模型将使用这些标注信息进行参数学习,以实现对有方向边界框目标的准确检测。 YOLOv5-OBB数据集可以包含各种不同种类的物体,比如车辆、行人、交通标识等。通过大规模、多样化的数据集训练得到的YOLOv5-OBB模型可以在实际应用具有较好的目标检测效果,能够对有方向边界框目标进行高效、准确的识别和定位。 总而言之,YOLOv5-OBB数据集是用于训练和评估YOLOv5-OBB模型的一种包含有方向边界框标注的数据集,可以有效支持对有方向边界框目标的检测任务。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值