Esempio n. 1
0
 def __init__(self, width=640, height=480):
     self._width = width
     self._height = height
     self._win = 'test features'
     self._img = None
     self._seq = None
     self._feature = GestureFeatures()
     self._resample = GestureResample(16)
     self._refresh()
Esempio n. 2
0
class TestFeatures():

    def __init__(self, width=640, height=480):
        self._width = width
        self._height = height
        self._win = 'test features'
        self._img = None
        self._seq = None
        self._feature = GestureFeatures()
        self._resample = GestureResample(16)
        self._refresh()

    def run(self):
        cv2.namedWindow(self._win)
        cv2.moveWindow(self._win, 0, 0)
        cv2.setMouseCallback(self._win, self._on_mouse, 0)

        while True:
            cv2.imshow(self._win, self._img)
            key = cv2.waitKey(WAIT_TIME)
            if key == ord('q'):
                break
            elif key == ord('r'):
                self._refresh()
            elif key == ord('c'):
                if self._seq:
                    seq = np.array(self._seq)
                    seq  = self._resample.run(seq, True)
                    self._draw_seq(seq)
                    f1, f2, f3, f4 = self._feature.run(seq)
                    print f4
                    self._refresh()

                print 'asd'
            else:
                continue

    def _refresh(self):
        self._img = np.zeros((self._height, self._width, 3), np.uint8)
        self._seq = []

    def _on_mouse(self, event, x, y, flags, params):
        if event == cv2.EVENT_LBUTTONDOWN:
            pt = (x, y)
            self._seq.append(pt)
            cv2.circle(self._img, pt, RADIUS, COLOR, -1)

    def _draw_seq(self, seq):
        tmp = np.zeros((self._height, self._width, 3), np.uint8)
        for pt in seq.astype(np.int).tolist():
            cv2.circle(tmp, tuple(pt), 5, (255,0,0), -1)
        cv2.imshow(self._win, tmp)
        cv2.waitKey(200)