YOLO格式的DOTA遥感数据集
YOLO格式的DOTA遥感数据集
AI道路千万条,准备数据第一条
DOTA_split数据集下载
最近为了训练yolo4和5,下载了DOTA数据集来用,不过下载的数据需要处理后让它符合yolo的标注格式,折腾了一天终于搞定了数据集的问题,现在分享一下数据集的地址(我只处理了V1.0版的DOTA,V1.5的虽然也下载下来了,但是原封没动):
百度云:https://pan.baidu.com/s/1UX7oX3_x5CrP_SxSA7XKXQ
提取码: iw3w
里面有两个压缩包,其中DOTA_original是从DOTA官网下载的,DOTA_split是根据yolo要求处理过的;
- images/ 中的图片已经改为.jpg格式,且已裁剪,因为我在colab训练yolov5的时候用png会报错
- lables/ 中标注文件已经处理为yolo格式
lables/ 中标签示例:
关于YOLOV5的训练效果
我在Google coalb上训练的V5,可惜只用了一天就被限制使用GPU了
训练时设置的 size = 640*640,1500张train,500张val(使用了预训练权重)
训练了100个epoch的效果为:
mAP看着真的很一般,就是速度还可以。作者建议根据数据集的大小训练50-300 epochs,要不是资本主义限制了我的算力,giao
等gpu限制解除了,我再去试试yolov4