skywrace / drowsiness_detector

Go to a project
Toggle navigation Toggle navigation pinning
skywrace
adb94818 1 parent 478e331e

chore: Test selecting 'eye' and 'mouth' landmarks from predictor

Inline Side-by-side
Showing 1 changed file with 8 additions and 2 deletions
...@@ -64,8 +64,14 @@ def main(): ...@@ -64,8 +64,14 @@ def main():
64 shape = predictor(gray, rect) 64 shape = predictor(gray, rect)
65 shape = face_utils.shape_to_np(shape) 65 shape = face_utils.shape_to_np(shape)
66 66
67 - # Draw on our image, all the finded cordinate points (x,y) 67 + r_eye = shape[36:43]
68 - for (x, y) in shape: 68 + l_eye = shape[42:48]
69 + mouth = shape[48:69]
70 + for (x, y) in l_eye:
71 + cv2.circle(frame, (x, y), 2, (0, 255, 0), -1)
72 + for (x, y) in r_eye:
73 + cv2.circle(frame, (x, y), 2, (0, 255, 0), -1)
74 + for (x, y) in mouth:
69 cv2.circle(frame, (x, y), 2, (0, 255, 0), -1) 75 cv2.circle(frame, (x, y), 2, (0, 255, 0), -1)
70 76
71 both_eyes_closed = False 77 both_eyes_closed = False
......