Yolo训练自己的数据集时P、R、mAP等值均为0,解决方法(亲测有效)
在自己想用yolo训练自己的数据集时,有时候会自己标注10来张图片进行测试,但训练100个epoch各项数据仍然为0,这时就会怀疑是否是自己的数据量不够,还是模型错误,还是哪个步骤没弄对,这里有一个简单的排除方法,可以测试自己的数据质量是否达标。,重复2次及以上,至少就有30几张图片和标签了,然后再次训练,这时候P、R、mAP都有值,但很小。所以你就需要考虑数据质量的问题了,搜集不同的图片进行标注
·
在自己想用yolo训练自己的数据集时,有时候会自己标注10来张图片进行测试,但训练100个epoch各项数据仍然为0,这时就会怀疑是否是自己的数据量不够,还是模型错误,还是哪个步骤没弄对,这里有一个简单的排除方法,可以测试自己的数据质量是否达标。
1、首先检查自己的数据集是否为一下格式,注意:图片的大小应该是一样的,val里面的图片大于2
├─images
│ ├─train
│ └─val
└─labels
├─train
└─val
2、data.yaml数据格式为以下格式
path: ../datasets/coco8-seg # dataset root dir
train: images/train # train images (relative to 'path') 4 images
val: images/val # val images (relative to 'path') 4 images
test: # test images (optional)
# nc: 6 # 目标检测可能有
# Classes
names:
0: person
1: bicycle
2: car
3: motorcycle
4: airplane
5: bus
3、最重要的一步,将图片和标签复制,粘贴,重复2次及以上,至少就有30几张图片和标签了,然后再次训练,这时候P、R、mAP都有值,但很小。所以你就需要考虑数据质量的问题了,搜集不同的图片进行标注,然后再次进行训练(检测+分割)。
DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。
更多推荐
所有评论(0)