-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
65 lines (57 loc) · 2.37 KB
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
import cv2
import mediapipe as mp
import pyautogui
for i in range(1):
pyautogui.alert("Make sure you are in good lighting and using a high-res camera.\n"
"Enjoy, Pranav Asrani\n"
"P.S: Press q to exit")
try:
# noinspection PyUnresolvedReferences
cam = cv2.VideoCapture(0)
face_mesh = mp.solutions.face_mesh.FaceMesh(refine_landmarks=True)
screen_w, screen_h = pyautogui.size()
amplification_factor = 2
while True:
_, frame = cam.read()
# noinspection PyUnresolvedReferences
frame = cv2.flip(frame, 1)
# noinspection PyUnresolvedReferences
rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
output = face_mesh.process(rgb_frame)
landmark_points = output.multi_face_landmarks
frame_h, frame_w, _ = frame.shape
if landmark_points:
landmarks = landmark_points[0].landmark
for id, landmark in enumerate(landmarks[474:478]):
x = int(landmark.x * frame_w)
y = int(landmark.y * frame_h)
# noinspection PyUnresolvedReferences
cv2.circle(frame, (x, y), 3, (0, 255, 0))
if id == 1:
screen_x = screen_w * landmark.x
screen_y = screen_h * landmark.y
pyautogui.moveTo(screen_x, screen_y)
left = [landmarks[145], landmarks[159]]
"""
print("frame_w: {} ".format(frame_w))
print("frame_h: {} ".format(frame_h))
"""
if cv2.waitKey(1) == ord('q'):
break
for landmark in left:
x = int(landmark.x * frame_w)
y = int(landmark.y * frame_h)
# noinspection PyUnresolvedReferences
cv2.circle(frame, (x, y), 3, (0, 255, 255))
if (left[0].y - left[1].y) < 0.009:
pyautogui.click()
print("click")
pyautogui.sleep(0.1)
# noinspection PyUnresolvedReferences
cv2.imshow('Iris Mouse', frame)
# noinspection PyUnresolvedReferences
cv2.waitKey(1)
except pyautogui.FailSafeException as e:
pyautogui.alert("do not move iris/mouse to edge of screen. may result in infinite loop. instead tilt head so that "
"both "
"irises stay within the frame")