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)))