MSCOCO数据库：怎么提取某一种类标签的图像

#导入数据库
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()

---------------------------------结果---------------------------------------------