在自己想用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都有值,但很小。所以你就需要考虑数据质量的问题了,搜集不同的图片进行标注,然后再次进行训练(检测+分割)。
在这里插入图片描述

Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐