Как я могу загрузить определенную часть набора данных 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);