87 lines
4.4 KiB
Python
87 lines
4.4 KiB
Python
import math
|
|
|
|
import cv2
|
|
import mediapipe as mp
|
|
import pyautogui
|
|
|
|
|
|
class Hands_AI:
|
|
def __init__(self):
|
|
self.cap = cv2.VideoCapture(0)
|
|
pyautogui.PAUSE = 0.01
|
|
self.mpHands = mp.solutions.hands
|
|
self.hands = self.mpHands.Hands(max_num_hands=1, min_tracking_confidence=0.95, min_detection_confidence=0.90)
|
|
self.mpDraw = mp.solutions.drawing_utils
|
|
self.mov_dis = 0.2479
|
|
self.click_dis = 0.2823
|
|
self.sensitivity = 3.5
|
|
self.x, self.y = None, None
|
|
self.click = False
|
|
|
|
def get_distance(self, first, second, height, width):
|
|
dist_x = (self.results.multi_hand_landmarks[0].landmark[first].x -
|
|
self.results.multi_hand_landmarks[0].landmark[
|
|
second].x) * width
|
|
dist_y = (self.results.multi_hand_landmarks[0].landmark[first].y -
|
|
self.results.multi_hand_landmarks[0].landmark[
|
|
second].y) * height
|
|
return math.sqrt(abs(dist_x ** 2 + dist_y ** 2))
|
|
|
|
def run(self, q):
|
|
while True:
|
|
success, img = self.cap.read()
|
|
self.h, self.w, self.c = img.shape
|
|
imgRGB = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
|
|
self.results = self.hands.process(imgRGB)
|
|
if self.results.multi_hand_landmarks:
|
|
dist_palm = round(self.get_distance(0, 5, self.h, self.w))
|
|
if not self.x and not self.y:
|
|
self.x, self.y = self.results.multi_hand_landmarks[0].landmark[4].x * self.w, \
|
|
self.results.multi_hand_landmarks[0].landmark[4].y * self.h
|
|
dis_1 = self.get_distance(12, 8, self.h, self.w)
|
|
cv2.putText(img,
|
|
f"""MOVE: {round(dis_1)}/{round(self.mov_dis * dist_palm, 2)} - {"true" if dis_1 < self.mov_dis * dist_palm else "false"}""",
|
|
(0, 15),
|
|
cv2.FONT_HERSHEY_PLAIN, 1.5, (255, 0, 255), thickness=2)
|
|
cv2.putText(img,
|
|
f"""CLICK: {round(self.get_distance(5, 3, self.h, self.w))}/{round(self.click_dis * dist_palm, 2)} - {"true" if self.get_distance(5, 3, self.h, self.w) < self.click_dis * dist_palm else "false"}""",
|
|
(0, 40),
|
|
cv2.FONT_HERSHEY_PLAIN, 1.5, (255, 0, 255), thickness=2)
|
|
cv2.putText(img,
|
|
f"""DRAG: {"true" if dis_1 < self.mov_dis * dist_palm and self.get_distance(5, 3, self.h, self.w) < self.click_dis * dist_palm else "false"}""",
|
|
(0, 65),
|
|
cv2.FONT_HERSHEY_PLAIN, 1.5, (255, 0, 255), thickness=2)
|
|
for handLms in self.results.multi_hand_landmarks:
|
|
self.mpDraw.draw_landmarks(img, handLms, self.mpHands.HAND_CONNECTIONS)
|
|
if dis_1 < self.mov_dis * dist_palm and self.get_distance(5, 3, self.h,
|
|
self.w) < self.click_dis * dist_palm:
|
|
pyautogui.mouseDown()
|
|
pyautogui.moveRel(
|
|
-(self.results.multi_hand_landmarks[0].landmark[8].x * self.w - self.x) * self.sensitivity,
|
|
(self.results.multi_hand_landmarks[0].landmark[8].y * self.h - self.y) * self.sensitivity,
|
|
duration=0.001)
|
|
elif dis_1 < self.mov_dis * dist_palm:
|
|
pyautogui.mouseUp()
|
|
pyautogui.moveRel(
|
|
-(self.results.multi_hand_landmarks[0].landmark[8].x * self.w - self.x) * self.sensitivity,
|
|
(self.results.multi_hand_landmarks[0].landmark[8].y * self.h - self.y) * self.sensitivity,
|
|
duration=0.001)
|
|
|
|
elif self.get_distance(5, 3, self.h, self.w) < self.click_dis * dist_palm and not self.click:
|
|
pyautogui.mouseUp()
|
|
pyautogui.click()
|
|
self.click = True
|
|
else:
|
|
pyautogui.mouseUp()
|
|
self.click = False
|
|
self.x, self.y = self.results.multi_hand_landmarks[0].landmark[8].x * self.w, \
|
|
self.results.multi_hand_landmarks[0].landmark[8].y * self.h
|
|
else:
|
|
self.x, self.y = None, None
|
|
cv2.imshow("Image", img)
|
|
cv2.waitKey(1)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
ai = Hands_AI()
|
|
ai.run("X") |