MSCOCO database: how to extract a certain kind of label image

#导入数据库
coco = COCO('instances_val2017.json')

#获取所有的图像ID号
total_imgIds = coco.getImgIds()
#获取'狗'的标签ID
catIds = coco.getCatIds(['dog'])

#在所有图像里面找到狗的图像的ID号
imgIds = coco.getImgIds(total_imgIds,catIds)

#根据找到的ID号找到图像信息
img = coco.loadImgs(imgIds)

#根据图像信息加载一张图像
I = io.imread('%s/%s/%s'%('E:/[1]论文/数据库/MSCOCO','val2017',img[0]['file_name']))

#用scikit-image显示出来
plt.axis('off')
plt.imshow(I)
plt.show()

---------------------------------result---------------- -----------------------------

Guess you like

Origin blog.csdn.net/XLcaoyi/article/details/88807630