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

zhaosuyuan

已于 2022-09-06 09:53:08 修改

阅读量416

点赞数 1

分类专栏： data 文章标签： python 人工智能算法

于 2022-09-06 09:52:48 首次发布

本文链接：https://blog.csdn.net/zhaosuyuan/article/details/126718822

版权

data 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

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