handposedetect.py

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)