使用 matplotlib 按类别着色

Question

这是我用 coco json 和图像创建带注释的图像的代码。

I = io.imread('%s/images/%s'%(dataDir,img['file_name']))

axes = plt.gca()
axes.set_frame_on(False)
axes.set_xticks([]); axes.set_yticks([])

annpath = os.path.join(dataDir, 'AnnImages/')
if not os.path.exists(annpath):
 os.makedirs(annpath)
path=annpath+str(img['file_name'])[:-4]+'.jpg'

annotation_ids = coco.getAnnIds(imgIds=img['id'], catIds=catIds, iscrowd=None)
annotations = coco.loadAnns(annotation_ids)
fig = plt.figure(figsize=(16, 9), dpi=155.9)
plt.imshow(I, interpolation='nearest')
plt.axis('off')

coco.showAnns(annotations)
    
plt.savefig(path, bbox_inches='tight', pad_inches=0)

但它为每个对象以不同的颜色输出。

我想为每个类别着色。

你能帮助我吗？

我的注释类别： test ， test2