如何用dota数据集训练yolov5模型

DOTA数据集介绍:Dota数据集介绍
这里我就不赘述了,相关的信息可以查看上面的链接
DOTA在标注的时候,分obb和hbb两种标注方式,这里我们需要选用hbb的标注方式进行训练

第一步:切割

DOTA中的原始图像分辨率普遍较高,最高分辨率甚至可以达到20000×20000,因此在训练之前会进行图像切割,官方建议以1024×1024的窗口大小,200的滑动步长对DOTA原始图像进行切割。

第二步:转换

将DOTA标注的格式的hbb转化成YOLO标注格式的hbb

第三步:验证

画图标注转化后的hbb,检查是否正确

参考1:DOTA_devkit_YOLO
参考2:dota格式的txt文件转化为yolo格式的txt文件

猜你喜欢

转载自blog.csdn.net/qq_34885993/article/details/130078256