набор данных 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 . Посмотрите, чтобы получить более подробную информацию и весь код.