如何下载Coco Dataset的特定部分？

Posted 2021-04-11

我正在开发一种物体检测模型来检测使用YOLO的船只。我想使用COCO数据集。有没有办法只下载带有注释的图像？

答案

据我个人所知，如果你只是谈论COCO数据集，我认为他们没有“船只”的类别。他们最接近的类别是“船”。这是检查可用类别的链接：http://cocodataset.org/#overview

顺便说一下，船类内也有船只。

如果您只想选择特定COCO类别的图像，您可能想要做这样的事情（从COCO的官方演示中获取和编辑）：

# display COCO categories
cats = coco.loadCats(coco.getCatIds())
nms=[cat['name'] for cat in cats]
print('COCO categories: 
{}
'.format(' '.join(nms)))

# get all images containing given categories (I'm selecting the "bird")
catIds = coco.getCatIds(catNms=['bird']);
imgIds = coco.getImgIds(catIds=catIds);