可视化COCO数据集中的标注

可视化COCO数据集中的标注

code snippets

通过opencv和matplotlib两种方式实现交互刷新图像

按下键盘按键显示下一张图像和标注

主要用到的API如下
opencv函数:

• cv2.rectangle()
• cv2.cv2.putText()
• cv2.waitkey()

matplotlib函数:

• plt.Rectangle()
• ax.annotate()
• plt.waitforbuttonpress()

完整的代码实例

from pycocotools.coco import COCO
import skimage.io as io
import matplotlib.pyplot as plt
import pylab, os, cv2, shutil
import os.path as osp
import numpy as np

coco_classes=[
        'person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', 'train',
        'truck', 'boat', 'traffic_light', 'fire_hydrant', 'stop_sign',
        'parking_meter', 'bench', 'bird', 'cat', 'dog', 'horse', 'sheep',
        'cow', 'elephant', 'bear', 'zebra', 'giraffe', 'backpack', 'umbrella',
        'handbag', 'tie', 'suitcase', 'frisbee', 'skis', 'snowboard',
        'sports_ball', 'kite', 'baseball_bat', 'baseball_glove', 'skateboard',
        'surfboard', 'tennis_racket', 'bottle', 'wine_glass', 'cup', 'fork',
        'knife', 'spoon', 'bowl', 'banana', 'apple', 'sandwich', 'orange',
        'broccoli', 'carrot', 'hot_dog', 'pizza', 'donut', 'cake', 'chair',
        'cou