-
Notifications
You must be signed in to change notification settings - Fork 0
/
hand_detect.py
45 lines (38 loc) · 1.49 KB
/
hand_detect.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
#!/usr/bin/env python3
"""
@Filename: hand_tacker.py
@Author: dulanj
@Time: 2021-09-23 15.44
"""
import cv2
import mediapipe as mp
from hands import Hands
class HandDetector():
def __init__(self, draw=True):
self.draw = draw
self.mp_drawing = mp.solutions.drawing_utils
self.mp_drawing_styles = mp.solutions.drawing_styles
self.mp_hands = mp.solutions.hands
def detect(self, image):
hands = self.mp_hands.Hands(
min_detection_confidence=0.5,
min_tracking_confidence=0.5)
image = cv2.cvtColor(cv2.flip(image, 1), cv2.COLOR_BGR2RGB)
# To improve performance, optionally mark the image as not writeable to
# pass by reference.
image.flags.writeable = False
results = hands.process(image)
hands_info = Hands(results)
if self.draw:
# Draw the hand annotations on the 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:
self.mp_drawing.draw_landmarks(
image,
hand_landmarks,
self.mp_hands.HAND_CONNECTIONS,
self.mp_drawing_styles.get_default_hand_landmarks_style(),
self.mp_drawing_styles.get_default_hand_connections_style())
return image, hands_info