1、功能
当我们在网上下载或者自己转换coco数据集时,想要验证转换或者下载的数据集标签是否正确。可以将标注框在原图上进行绘制,从而可以直观看到标签是否正确。
2、代码实现
import os
from pycocotools.coco import COCO
from skimage import io
from matplotlib import pyplot as plt
json_file = 'C:/Users/37449/Desktop/coco/annotations/instances_val.json' #json文件路径
dataset_dir = 'C:/Users/37449/Desktop/coco/val/' #json文件对应的图片路径
coco = COCO(json_file)
catIds = coco.getCatIds(catNms=[]) #[]表示所有标签都显示,也可指定某个标签进行显示
imgIds = coco.getImgIds(catIds=catIds ) # 图片id,许多值
for i in range(len(imgIds)):img = coco.loadImgs(imgIds[i])[0]I = io.imread(dataset_dir + img['file_name'])plt.axis('off')plt.imshow(I)annIds = coco.getAnnIds(imgIds=img['id'], catIds=catIds, iscrowd=None)anns = coco.loadAnns(annIds)coco.showAnns(anns)plt.show()