agrobot_base/Python/hand-recognize/hand-recognize.py

39 lines
1.1 KiB
Python

import cv2
import mediapipe as mp
# Inicializa o módulo MediaPipe Hands
mp_hands = mp.solutions.hands
hands = mp_hands.Hands()
# Inicializa a captura de vídeo
cap = cv2.VideoCapture(0)
while cap.isOpened():
# Captura o frame da câmera
ret, frame = cap.read()
if not ret:
break
# Converte o frame para RGB e detecta as mãos
rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
results = hands.process(rgb_frame)
# Desenha os pontos da detecção das mãos, se houverem
if results.multi_hand_landmarks:
for hand_landmarks in results.multi_hand_landmarks:
for idx, landmark in enumerate(hand_landmarks.landmark):
h, w, c = frame.shape
cx, cy = int(landmark.x * w), int(landmark.y * h)
cv2.circle(frame, (cx, cy), 5, (255, 0, 0), cv2.FILLED)
# Mostra o frame com os pontos das mãos
cv2.imshow('MediaPipe Hands', frame)
# Verifica se o usuário pressionou 'q' para sair
if cv2.waitKey(10) & 0xFF == ord('q'):
break
# Libera a captura de vídeo e fecha as janelas
cap.release()
cv2.destroyAllWindows()