Skip to content

Instantly share code, notes, and snippets.

@smith0022
Created August 12, 2021 11:22
Show Gist options
  • Save smith0022/297b039d09046030d1e89aed9c583834 to your computer and use it in GitHub Desktop.
Save smith0022/297b039d09046030d1e89aed9c583834 to your computer and use it in GitHub Desktop.
a module for tracking and can be used for various other purposes
import cv2 as c
import mediapipe as mp
import time
class HandDetect() :
def __init__(self, mode=False, maxhands=2, detectC=0.5, trackC=0.5):
self.mode = mode
self.maxhands = maxhands
self.detectC = detectC
self.trackC = trackC
self.mpHands = mp.solutions.hands
self.hands = self.mpHands.Hands(self.mode, self.maxhands, self.detectC, self.trackC)
self.mpDraw = mp.solutions.drawing_utils
def findhands(self,img,draw = True):
Rgb = c.cvtColor(img, c.COLOR_BGR2RGB)
self.result = self.hands.process(Rgb)
# print(result.multi_hand_landmarks)
if self.result.multi_hand_landmarks:
for HandsLms in self.result.multi_hand_landmarks:
if draw:
self.mpDraw.draw_landmarks(img, HandsLms, self.mpHands.HAND_CONNECTIONS)
return img
def findPosition(self,img,handNo=0,draw=True):
lmlist = []
if self.result.multi_hand_landmarks:
myhand=self.result.multi_hand_landmarks[handNo]
for id, lm in enumerate(myhand.landmark):
# print(id,lm)
h, w, cen = img.shape
cx, cy = int(lm.x * w), int(lm.y * h)
#print(id, cx, cy)
lmlist.append([id,cx, cy])
#if id in (4, 8, 16, 12, 20) and draw==True:
#c.circle(img, (cx, cy), 15, (255, 0, 255), c.FILLED)return
return lmlist
def main():#to write from this everytime for usuage
cap = c.VideoCapture(0)
ctime = 0
ptime = 0
HDT=HandDetect()
while True:
success, img = cap.read()
img = HDT.findhands(img,False)#put false fr noy showing the lines
HDT.findPosition(img)#false for not shoing dots as well
ctime = time.time()
fps = 1/(ctime-ptime)
ptime = ctime
c.putText(img, str(int(fps)), (10, 70), c.FONT_HERSHEY_PLAIN, 3, (255, 0, 0), 3)
c.imshow("Image", img)
c.waitKey(1)
if __name__ == "__main__":
main()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment