visualize_pixels.py 文件源码

def __init__(self, mask, imsizex, imsizey, dpi=100):

        # convert mask to x,y points
        points = map(list, zip(*reversed(map(list, np.where(mask)))))

        # make subdivision 
        rect = (0, 0, imsizex, imsizey)
        self.subdiv = cv2.Subdiv2D(rect)
        self.subdiv.insert(points)

        # initialize figure
        self.fig = plt.figure("Image Stream",
                              figsize=(imsizex / dpi, imsizey / dpi),
                              dpi=dpi)

        # initialize ax
        self.ax = self.fig.add_axes([0, 0, 1, 1], frame_on=False)
        self.ax.xaxis.set_visible(False)
        self.ax.yaxis.set_visible(False)

        # initialize image
        self.im = self.ax.imshow(np.zeros((imsizey, imsizex)),
                                 interpolation='none',
                                 animated=True)

        self.image = None
        # start figure
        self.fig.show()
        self.fig.canvas.draw()