前言:
在使用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格式的图片就跳过的这个判断,真坑啊!
解决方法就是注释掉就行,大功告成!