如何从 MSCOCO 数据集中获取人员类别和分段？

Question

我只想从 COCO 数据集中下载人员类别和二进制分割。我该怎么做？

Answer 1

使用pycocotools。

• 导入库

  from pycocotools.coco import COCO
  

• 加载coco注解的json文件

  coco = COCO('/home/office/cocoDataset/annotations/instances_train2017.json')
  

• 获取coco数据集的类别ID

  category_ids = coco.getCatIds(catNms=['person'])
  

• 获取单个图像的注释

  annotations = coco.getAnnIds(imgIds=image_id, catIds=category_ids, iscrowd=False)
  

• 这里每个人都有不同的注释，第i个人的注释annotation[i]因此合并所有注释并保存

  if annotations:
    mask = coco.annToMask(annotations[0])
    for i in range(len(annotations)):
      mask |= coco.annToMask(annotations[i])
    mask = mask * 255
    im = Image.fromarray(mask)
    im.save('~/mask_name.png')