Spaces:
Running
Running
| import mediapipe as mp | |
| import cv2 | |
| from utils import read_n_resize | |
| def mp_hand_pose_detect_fn(image, min_detect_conf=0.5): | |
| mp_drawing = mp.solutions.drawing_utils | |
| mp_drawing_styles = mp.solutions.drawing_styles | |
| mp_hands = mp.solutions.hands | |
| with mp_hands.Hands( | |
| static_image_mode=True, | |
| max_num_hands=5, | |
| min_detection_confidence=min_detect_conf | |
| ) as hands: | |
| image = cv2.flip(read_n_resize(image, read=False), 1) | |
| results = hands.process(image) | |
| image_height, image_width, _ = image.shape | |
| annotated_image = image.copy() | |
| for hand_landmarks in results.multi_hand_landmarks: | |
| mp_drawing.draw_landmarks( | |
| annotated_image, | |
| hand_landmarks, | |
| mp_hands.HAND_CONNECTIONS, | |
| mp_drawing_styles.get_default_hand_landmarks_style(), | |
| mp_drawing_styles.get_default_hand_connections_style()) | |
| return cv2.flip(annotated_image, 1) | |