SeguimientoManos.py

#-----------------------Importacion de librerias ------------------------------------

import math
import cv2
import mediapipe as mp
import time

# -------------------------- Creamos la clase -----------------------------------

class DetectorManos():
    # ----------------- Inicalizar parametros de deteccion -----------------------
    def __init__(self, mode = False, maxManos = 2, confDeteccion =  0.5, confSegui = 0.5):
        self.mode = mode
        self.maxManos = maxManos
        self.confDeteccion = confDeteccion
        self.confSegui = confSegui

        # ----------------- Crear objetos para detectar manos y dibujarlas ------------------
        self.mpmanos = mp.solutions.hands
        self.manos = self.mpmanos.Hands(self.mode, self.maxManos, self.confDeteccion, self.confSegui)
        self.dibujo = mp.solutions.drawing_utils
        self.tip = [4,8,12,14,16,20]

    # ----------------- Funcion para encontrar las manos -----------------------
    def encontrarManos(self, frame, dibujar = True):
        imgcolor = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
        self.resultados = self.manos.process(imgcolor)
        if self.resultados.multi_hand_landmarks:
            for mano in self.resultados.multi_hand_landmarks:
                if dibujar:
                    self.dibujo.draw_landmarks(frame, mano, self.mpmanos.HAND_CONNECTIONS)
        return frame

    # ----------------- Funcion para encontrar la posicion -----------------------
    def encontrarPosicion(self, frame, ManoNum = 0, dibujar = True):
        xlista = []
        ylista = []
        bbox = []
        self.lista = []
        if self.resultados.multi_hand_landmarks:
            miMano = self.resultados.multi_hand_landmarks[ManoNum]
            for id, lm in enumerate(miMano.landmark):
                alto, ancho, c = frame.shape
                cx, cy = int(lm.x * ancho), int(lm.y * alto)
                xlista.append(cx)
                ylista.append(cy)
                self.lista.append([id, cx, cy])
                if dibujar:
                    cv2.circle(frame, (cx, cy), 5, (0,0,0), cv2.FILLED)
            
            xmin, xmax = min(xlista), max(xlista)
            ymin, ymax = min(ylista), max(ylista)
            bbox = xmin, ymin, xmax, ymax
            if dibujar:
                cv2.rectangle(frame, (xmin - 20, ymin - 20), (xmax + 20, ymax + 20), (0,255,0),2)
        return self.lista, bbox

    # ----------------- Funcion para detectar dedos arriba -----------------------
    def dedosarriba(self):
        dedos = []
        if self.lista[self.tip[0]][1] > self.lista[self.tip[0]-1][1]:
            dedos.append(1)
        else:
            dedos.append(0)

        for id in range(1,5):
            if self.lista[self.tip[id]][2] > self.lista[self.tip[id]-2][2]:
                dedos.append(1)
            else:
                dedos.append(0)
        return dedos

    # ----------------- Funcion para detectar la distancia entre dedos -----------------------
    def distancia(self, p1, p2, frame, dibujar=True, r = 15, t = 3):
        x1, y1 = self.lista[p1][1:]
        x2, y2 = self.lista[p2][1:]
        cx, cy = (x1 + x2) //2, (y1 + y2) //2
        if dibujar:
            cv2.line(frame, (x1, y1), (x2, y2), (0,0,255),t)
            cv2.circle(frame, (x1, y1), r, (0,0,255), cv2.FILLED)
            cv2.circle(frame, (x2, y2), r, (0,0,255), cv2.FILLED)
            cv2.circle(frame, (cx, cy), r, (0,0,255), cv2.FILLED)
        lenght = math.hypot(x2-x1, y2-y1)
        return lenght, frame, [x1, y1, x2, y2, cx, cy]

# --------------------------------- Funcion Principal --------------------------------------------

def main():
    ptiempo = 0
    ctiempo = 0

    #------------------- Leemos la camara web -----------------------
    cap = cv2.VideoCapture(0)
    #------------------- Creamos el objeto -----------------------
    detector = DetectorManos()
    #------------------- Realizamos la deteccion de manos -----------------------
    while True:
        ret, frame = cap.read()
        frame = detector.encontrarManos(frame)
        lista, bbox = detector.encontrarPosicion(frame)

        #------------------- Mostramos los fps -----------------------
        ctiempo = time.time()
        fps = 1 / (ctiempo - ptiempo)
        ptiempo = ctiempo

        cv2.putText(frame, str(int(fps)), (10,70), cv2.FONT_HERSHEY_PLAIN, 3, (255, 0, 255), 3)

        cv2.imshow("Detector de manos", frame)
        k = cv2.waitKey(1)

        if k == 27:
            break
    cap.release()
    cv2.destroyAllWindows()

if __name__ == '__main__':
    main()