python - 如何下载Coco数据集的特定部分？

Question

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

Answer 1

据我个人所知，如果您仅谈论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: \n{}\n'.format(' '.join(nms)))

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