-
Notifications
You must be signed in to change notification settings - Fork 0
/
draw.py
69 lines (56 loc) · 2.9 KB
/
draw.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
66
67
68
import mediapipe as mp
import cv2
import gesture
class HandTracking:
def __init__(self, image) -> None:
self.mp_hands = mp.solutions.hands
self.mp_drawing = mp.solutions.drawing_utils
self.draw = False
self.count = 0
self.new_points = []
self.all_points = []
self.image = image
def read_video_capture(self) -> None:
self.image = cv2.cvtColor(cv2.flip(self.image, 1), cv2.COLOR_BGR2RGB)
self.results = self.hands.process(self.image)
self.image.flags.writeable = True
self.image = cv2.cvtColor(self.image, cv2.COLOR_RGB2BGR)
return True
def draw_hand_connections(self) -> None:
if self.results.multi_hand_landmarks:
for self.hand_landmarks in self.results.multi_hand_landmarks:
self.mp_drawing.draw_landmarks(self.image, self.hand_landmarks, self.mp_hands.HAND_CONNECTIONS)
def update_draw(self) -> None:
if gesture.is_closed(self.results):
self.count += 1
if self.count == 30:
self.draw = not(self.draw)
self.count = 0
def actual_coordinates(self) -> None:
index_finger_coords = self.hand_landmarks.landmark[8]
image_height, image_width, _ = self.image.shape
self.x, self.y = index_finger_coords.x, index_finger_coords.y
def draw_line(self) -> None:
if self.results.multi_hand_landmarks is not None and self.draw:
self.actual_coordinates()
self.new_points.append((self.x, self.y))
for points in self.all_points:
for i in range(len(points) - 1):
cv2.line(self.image, (points[i][0], points[i][1]), (points[i + 1][0], points[i + 1][1]), color = (255, 255, 0), thickness = 10)
for i in range(len(self.new_points) - 1):
cv2.line(self.image, (self.new_points[i][0], self.new_points[i][1]), (self.new_points[i + 1][0], self.new_points[i + 1][1]), color = (255, 255, 0), thickness = 10)
else:
for points in self.all_points:
for i in range(len(points) - 1):
cv2.line(self.image, (points[i][0], points[i][1]), (points[i + 1][0], points[i + 1][1]), color = (255, 255, 0), thickness = 10)
self.all_points.append(self.new_points)
self.new_points = []
def start_drawing(self) -> None:
with self.mp_hands.Hands(min_detection_confidence=0.5, min_tracking_confidence=0.5) as self.hands:
if not(self.read_video_capture()):
return None
self.draw_hand_connections()
if self.results.multi_hand_landmarks is not None:
self.actual_coordinates()
return gesture.is_closed(self.results), self.x, self.y, self.image
return False, 0, 0, self.image