from __future__ import absolute_import from __future__ import division from __future__ import print_function from __future__ import unicode_literals import copy import cPickle as pickle import logging import numpy as np import os import scipy.sparse # Must happen before importing COCO API (which imports matplotlib) import detectron.utils.env as envu envu.set_up_matplotlib() # COCO API from pycocotools import mask as COCOmask from pycocotools.coco import COCO from detectron.core.config import cfg from detectron.utils.timer import Timer import detectron.datasets.dataset_catalog as dataset_catalog import detectron.utils.boxes as box_utils logger = logging.getLogger(__name__) class JsonDataset(object): """A class representing a COCO json dataset.""" def __init__(self, name):
from __future__ import print_function from __future__ import unicode_literals import cv2 import numpy as np import os import pycocotools.mask as mask_util from detectron.utils.colormap import colormap import detectron.utils.env as envu import detectron.utils.keypoints as keypoint_utils # Matplotlib requires certain adjustments in some environments # Must happen before importing matplotlib envu.set_up_matplotlib() import matplotlib.pyplot as plt from matplotlib.patches import Polygon plt.rcParams['pdf.fonttype'] = 42 # For editing in Adobe Illustrator _GRAY = (218, 227, 218) _GREEN = (18, 127, 15) _WHITE = (255, 255, 255) def kp_connections(keypoints): kp_lines = [ [keypoints.index('left_eye'), keypoints.index('right_eye')], [keypoints.index('left_eye'), keypoints.index('nose')],