最近我需要使用pycocotools根据图片id获取categories。但是网上只有mscoco官方例程中的根据catids获取image,所以我从cocoapi的源码里找到了解决办法,记录如下。
实现方法
coco = COCO(labels_file)
imgIds = coco.getImgIds()
# 此处imgIds代表了所需查阅的图片id
annIds = coco.getAnnIds(imgIds = imgIds[0])
anns = coco.loadAnns(annIds)
labels = []
for ann in anns:
labels.append(ann['id'])
# labels即为该imgid对应的labels
print(labels)