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()
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)