coco提取每个类别的图片数量和框bbox数量

coco提取每个类别的图片数量和框bbox数量

from pycocotools.coco import COCO
annFile='train.json'
coco=COCO(annFile)
# display COCO categories and supercategories
cats = coco.loadCats(coco.getCatIds()) 
cat_nms=[cat['name'] for cat in cats] 
#print('COCO categories: \n{}\n'.format(' '.join(cat_nms)))
print("{:<15} {:<5}     {:<10}".format('classname', 'imgnum', 'bboxnum'))
print('---------------------------------')
for cat_name in cat_nms:
    catId = coco.getCatIds(catNms=[cat_name])
    imgId = coco.getImgIds(catIds=catId)
    annId = coco.getAnnIds(imgIds=imgId, catIds=catId, iscrowd=None)                  
    print("{:<15} {:<6d}     {:<10d}".format(cat_name, len(imgId), len(annId)))