Handtrack_module.py

import cv2
import mediapipe as mp
import time

class handdetector():
    def __init__(self,mode=False,maxhands=2,modelC=1,detectionCon=0.5,trackCon=0.5):
        self.mode=mode
        self.maxHands=maxhands
        self.detectionCon=detectionCon
        self.modelC=modelC
        self.trackCon=trackCon
        self.mpHands=mp.solutions.hands
        self.hands=self.mpHands.Hands(self.mode,self.maxHands,self.modelC,self.detectionCon,self.trackCon)
        self.mpDraw = mp.solutions.drawing_utils
        self.tipIds = [4,8,12,16,20]
    def findHands(self,img,draw=False): 
        imgRGB=cv2.cvtColor(img,cv2.COLOR_BGR2RGB)
        self.results=self.hands.process(imgRGB)
        # print(results.multi_hand_landmarks)
        if self.results.multi_hand_landmarks:
            for handLms in self.results.multi_hand_landmarks:
                
                if(draw==True):
                    self.mpDraw.draw_landmarks(img,handLms,self.mpHands.HAND_CONNECTIONS)
        return img
    def findposition(self,img,handNo=0,draw=False):
        self.lmlist= []
        if self.results.multi_hand_landmarks:
            myhand=self.results.multi_hand_landmarks[handNo]
            for id,lm in enumerate(myhand.landmark):
                # print(id,lm)
                h,w,c=img.shape
                cx,cy=int(lm.x*w),int(lm.y*h)
                # print(id,cx,cy)
                self.lmlist.append([id,cx,cy])
                if draw:
                    cv2.circle(img,(cx,cy),15,(255,0,255),cv2.FILLED)
        return self.lmlist
    def fingersUp(self):
        fingers = []
        if self.lmlist[self.tipIds[0]][1]>self.lmlist[self.tipIds[0]-1][1]:
            fingers.append(1)
        else:
            fingers.append(0)
        for id in range(1,5):
            if self.lmlist[self.tipIds[id]][2]<self.lmlist[self.tipIds[id]-2][2]:
                fingers.append(1)
            else:
                fingers.append(0)
        return fingers

def main():
    
    cap=cv2.VideoCapture(0)

    Ptime=0
    Ctime=0
    detector=handdetector()
    while True:
        success,img =cap.read()
        img=detector.findHands(img) 
        lmList=detector.findposition(img)
        if(len(lmList)!=0):
            print(lmList[4])
        Ctime =time.time()
        fps=1/(Ctime-Ptime)
        Ptime=Ctime

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

        cv2.imshow("Image",img)
        cv2.waitKey(1)
     

if __name__ == "__main__":
    main()