39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
import cv2
|
|
import mediapipe as mp
|
|
import simple_gestures_lib as sgest
|
|
|
|
mp_drawing = mp.solutions.drawing_utils
|
|
mp_hands = mp.solutions.hands
|
|
from math import sqrt
|
|
|
|
|
|
|
|
hands = mp_hands.Hands(
|
|
min_detection_confidence=0.5, min_tracking_confidence=0.5)
|
|
cap = cv2.VideoCapture(0)
|
|
while cap.isOpened():
|
|
success, image = cap.read()
|
|
image = cv2.cvtColor(cv2.flip(image, 1), cv2.COLOR_BGR2RGB)
|
|
image.flags.writeable = False
|
|
results = hands.process(image)
|
|
|
|
image.flags.writeable = True
|
|
image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
|
|
if results.multi_hand_landmarks:
|
|
for hand_landmarks in results.multi_hand_landmarks:
|
|
mp_drawing.draw_landmarks(
|
|
image, hand_landmarks, mp_hands.HAND_CONNECTIONS)
|
|
if cv2.waitKey(33) == ord('s'):
|
|
if results.multi_hand_landmarks:
|
|
for hand_landmarks in results.multi_hand_landmarks:
|
|
print(sgest.check_index_finger(hand_landmarks))
|
|
print(sgest.check_middle_finger(hand_landmarks))
|
|
print(sgest.check_ring_finger(hand_landmarks))
|
|
print(sgest.check_pinky_finger(hand_landmarks))
|
|
|
|
cv2.imshow('MediaPipe Hands', image)
|
|
if cv2.waitKey(5) & 0xFF == 27:
|
|
break
|
|
hands.close()
|
|
cap.release()
|