COCO набор данных количество изображений на классы

Question

Я вижу, что COCO2017 имеет 80 классов 118k обучения и 5k набора данных проверки (122k изображений). У меня есть вопрос здесь. Соответствует ли количество изображений на классы (1525 изображений на класс) ~ 122k / 80?

Answer 1

набор данных COCO не является равномерно распределенным набором данных, т. Е. Все классы не имеют одинаковое количество изображений. Итак, позвольте мне показать вам способ узнать количество изображений в любом классе, которым вы пользуетесь sh.

Я использую PyCoco API для работы с набором данных COCO. Давайте выясним количество изображений в классе 'person' набора данных COCO. Вот код, отфильтровывающий любой класс из набора данных COCO:

# Define the class (out of the 80 COCO classes)
filterClasses = ['person']

# Fetch class IDs only corresponding to the filterClasses
catIds = coco.getCatIds(catNms=filterClasses) 

# Get all images containing the above Category IDs
imgIds = coco.getImgIds(catIds=catIds)
print("Number of images containing the class:", len(imgIds))

Там мы получаем количество изображений, соответствующих 'person' в наборе данных!

, которые я недавно получил написал целый пост , исследуя набор данных COCO . Посмотрите, чтобы получить более подробную информацию и весь код.