COCO API 显示图片的分割效果

from pycocotools.coco import COCO
import numpy as np
import skimage.io as io
import matplotlib.pyplot as plt
import pylab

pylab.rcParams['figure.figsize'] = (8.0, 10.0)

dataDir = '文件夹路径'
dataType = 'val2017'
annFile = '{}/annotations/instances_{}.json'.format(dataDir, dataType)

# I为图片具体位置
I = io.imread(dataDir + '/annotations/val2017/' + '000000324158.jpg')

coco = COCO(annFile)

plt.imshow(I)
plt.axis('off')
annIds = coco.getAnnIds(imgIds=324158, iscrowd=None)
# print(annIds)
anns = coco.loadAnns(annIds)
# print(anns)
coco.showAnns(anns)
plt.show()