python-pose-estimation/PoseModule.py at main · Rahul-learner/python-pose-estimation · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
importtime
importmediapipeasmp
importcv2


classposeDetector():

def__init__(self, mode=False, upBody=False, smooth=True,
detectionCon=0.5, trackCon=0.5):
self.mode=mode
self.upBody=upBody
self.smooth=smooth
self.detectionCon=detectionCon
self.trackCon=trackCon


self.mpDraw=mp.solutions.drawing_utils
self.mpPose=mp.solutions.pose
self.pose=self.mpPose.Pose(self.mode,self.upBody,self.smooth,
self.detectionCon,self.trackCon)

deffindPose(self, img, draw=True):
imgRGB=cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
self.results=self.pose.process(imgRGB)
ifself.results.pose_landmarks:
ifdraw:
self.mpDraw.draw_landmarks(img, self.results.pose_landmarks,
self.mpPose.POSE_CONNECTIONS)
returnimg

deffindPosition(self, img, draw=True):
lmList= []
ifself.results.pose_landmarks:
forid, lminenumerate(self.results.pose_landmarks.landmark):
h, w, c=img.shape
#print(id, lm)
cx, cy=int(lm.x*w), int(lm.y*h)
lmList.append([id, cx, cy])
ifdraw:
cv2.circle(img, (cx, cy), 5, (255, 0, 0), cv2.FILLED)
returnlmList


defmain():
cap=cv2.VideoCapture('3.mp4')
pTime=0
detector=poseDetector()
whileTrue:
success, img=cap.read()
img=detector.findPose(img)
lmList=detector.findPosition(img,draw=False)
iflen(lmList) !=0:
print(lmList[14])
cv2.circle(img, (lmList[14][1], lmList[14][2]), 10, (0, 0, 255), cv2.FILLED)

cTime=time.time()
fps=2.5/ (cTime-pTime)
pTime=cTime

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

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

if__name__=="__main__":
main()
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
importtime
importmediapipeasmp
importcv2


classposeDetector():

def__init__(self, mode=False, upBody=False, smooth=True,
detectionCon=0.5, trackCon=0.5):
self.mode=mode
self.upBody=upBody
self.smooth=smooth
self.detectionCon=detectionCon
self.trackCon=trackCon


self.mpDraw=mp.solutions.drawing_utils
self.mpPose=mp.solutions.pose
self.pose=self.mpPose.Pose(self.mode,self.upBody,self.smooth,
self.detectionCon,self.trackCon)

deffindPose(self, img, draw=True):
imgRGB=cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
self.results=self.pose.process(imgRGB)
ifself.results.pose_landmarks:
ifdraw:
self.mpDraw.draw_landmarks(img, self.results.pose_landmarks,
self.mpPose.POSE_CONNECTIONS)
returnimg

deffindPosition(self, img, draw=True):
lmList= []
ifself.results.pose_landmarks:
forid, lminenumerate(self.results.pose_landmarks.landmark):
h, w, c=img.shape
#print(id, lm)
cx, cy=int(lm.x*w), int(lm.y*h)
lmList.append([id, cx, cy])
ifdraw:
cv2.circle(img, (cx, cy), 5, (255, 0, 0), cv2.FILLED)
returnlmList


defmain():
cap=cv2.VideoCapture('3.mp4')
pTime=0
detector=poseDetector()
whileTrue:
success, img=cap.read()
img=detector.findPose(img)
lmList=detector.findPosition(img,draw=False)
iflen(lmList) !=0:
print(lmList[14])
cv2.circle(img, (lmList[14][1], lmList[14][2]), 10, (0, 0, 255), cv2.FILLED)

cTime=time.time()
fps=2.5/ (cTime-pTime)
pTime=cTime

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

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

if__name__=="__main__":
main()
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
importtime
importmediapipeasmp
importcv2


classposeDetector():

def__init__(self, mode=False, upBody=False, smooth=True,
detectionCon=0.5, trackCon=0.5):
self.mode=mode
self.upBody=upBody
self.smooth=smooth
self.detectionCon=detectionCon
self.trackCon=trackCon


self.mpDraw=mp.solutions.drawing_utils
self.mpPose=mp.solutions.pose
self.pose=self.mpPose.Pose(self.mode,self.upBody,self.smooth,
self.detectionCon,self.trackCon)

deffindPose(self, img, draw=True):
imgRGB=cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
self.results=self.pose.process(imgRGB)
ifself.results.pose_landmarks:
ifdraw:
self.mpDraw.draw_landmarks(img, self.results.pose_landmarks,
self.mpPose.POSE_CONNECTIONS)
returnimg

deffindPosition(self, img, draw=True):
lmList= []
ifself.results.pose_landmarks:
forid, lminenumerate(self.results.pose_landmarks.landmark):
h, w, c=img.shape
#print(id, lm)
cx, cy=int(lm.x*w), int(lm.y*h)
lmList.append([id, cx, cy])
ifdraw:
cv2.circle(img, (cx, cy), 5, (255, 0, 0), cv2.FILLED)
returnlmList


defmain():
cap=cv2.VideoCapture('3.mp4')
pTime=0
detector=poseDetector()
whileTrue:
success, img=cap.read()
img=detector.findPose(img)
lmList=detector.findPosition(img,draw=False)
iflen(lmList) !=0:
print(lmList[14])
cv2.circle(img, (lmList[14][1], lmList[14][2]), 10, (0, 0, 255), cv2.FILLED)

cTime=time.time()
fps=2.5/ (cTime-pTime)
pTime=cTime

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

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

if__name__=="__main__":
main()
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
importtime
importmediapipeasmp
importcv2


classposeDetector():

def__init__(self, mode=False, upBody=False, smooth=True,
detectionCon=0.5, trackCon=0.5):
self.mode=mode
self.upBody=upBody
self.smooth=smooth
self.detectionCon=detectionCon
self.trackCon=trackCon


self.mpDraw=mp.solutions.drawing_utils
self.mpPose=mp.solutions.pose
self.pose=self.mpPose.Pose(self.mode,self.upBody,self.smooth,
self.detectionCon,self.trackCon)

deffindPose(self, img, draw=True):
imgRGB=cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
self.results=self.pose.process(imgRGB)
ifself.results.pose_landmarks:
ifdraw:
self.mpDraw.draw_landmarks(img, self.results.pose_landmarks,
self.mpPose.POSE_CONNECTIONS)
returnimg

deffindPosition(self, img, draw=True):
lmList= []
ifself.results.pose_landmarks:
forid, lminenumerate(self.results.pose_landmarks.landmark):
h, w, c=img.shape
#print(id, lm)
cx, cy=int(lm.x*w), int(lm.y*h)
lmList.append([id, cx, cy])
ifdraw:
cv2.circle(img, (cx, cy), 5, (255, 0, 0), cv2.FILLED)
returnlmList


defmain():
cap=cv2.VideoCapture('3.mp4')
pTime=0
detector=poseDetector()
whileTrue:
success, img=cap.read()
img=detector.findPose(img)
lmList=detector.findPosition(img,draw=False)
iflen(lmList) !=0:
print(lmList[14])
cv2.circle(img, (lmList[14][1], lmList[14][2]), 10, (0, 0, 255), cv2.FILLED)

cTime=time.time()
fps=2.5/ (cTime-pTime)
pTime=cTime

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

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

if__name__=="__main__":
main()
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
importtime
importmediapipeasmp
importcv2


classposeDetector():

def__init__(self, mode=False, upBody=False, smooth=True,
detectionCon=0.5, trackCon=0.5):
self.mode=mode
self.upBody=upBody
self.smooth=smooth
self.detectionCon=detectionCon
self.trackCon=trackCon


self.mpDraw=mp.solutions.drawing_utils
self.mpPose=mp.solutions.pose
self.pose=self.mpPose.Pose(self.mode,self.upBody,self.smooth,
self.detectionCon,self.trackCon)

deffindPose(self, img, draw=True):
imgRGB=cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
self.results=self.pose.process(imgRGB)
ifself.results.pose_landmarks:
ifdraw:
self.mpDraw.draw_landmarks(img, self.results.pose_landmarks,
self.mpPose.POSE_CONNECTIONS)
returnimg

deffindPosition(self, img, draw=True):
lmList= []
ifself.results.pose_landmarks:
forid, lminenumerate(self.results.pose_landmarks.landmark):
h, w, c=img.shape
#print(id, lm)
cx, cy=int(lm.x*w), int(lm.y*h)
lmList.append([id, cx, cy])
ifdraw:
cv2.circle(img, (cx, cy), 5, (255, 0, 0), cv2.FILLED)
returnlmList


defmain():
cap=cv2.VideoCapture('3.mp4')
pTime=0
detector=poseDetector()
whileTrue:
success, img=cap.read()
img=detector.findPose(img)
lmList=detector.findPosition(img,draw=False)
iflen(lmList) !=0:
print(lmList[14])
cv2.circle(img, (lmList[14][1], lmList[14][2]), 10, (0, 0, 255), cv2.FILLED)

cTime=time.time()
fps=2.5/ (cTime-pTime)
pTime=cTime

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

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

if__name__=="__main__":
main()
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
importtime
importmediapipeasmp
importcv2


classposeDetector():

def__init__(self, mode=False, upBody=False, smooth=True,
detectionCon=0.5, trackCon=0.5):
self.mode=mode
self.upBody=upBody
self.smooth=smooth
self.detectionCon=detectionCon
self.trackCon=trackCon


self.mpDraw=mp.solutions.drawing_utils
self.mpPose=mp.solutions.pose
self.pose=self.mpPose.Pose(self.mode,self.upBody,self.smooth,
self.detectionCon,self.trackCon)

deffindPose(self, img, draw=True):
imgRGB=cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
self.results=self.pose.process(imgRGB)
ifself.results.pose_landmarks:
ifdraw:
self.mpDraw.draw_landmarks(img, self.results.pose_landmarks,
self.mpPose.POSE_CONNECTIONS)
returnimg

deffindPosition(self, img, draw=True):
lmList= []
ifself.results.pose_landmarks:
forid, lminenumerate(self.results.pose_landmarks.landmark):
h, w, c=img.shape
#print(id, lm)
cx, cy=int(lm.x*w), int(lm.y*h)
lmList.append([id, cx, cy])
ifdraw:
cv2.circle(img, (cx, cy), 5, (255, 0, 0), cv2.FILLED)
returnlmList


defmain():
cap=cv2.VideoCapture('3.mp4')
pTime=0
detector=poseDetector()
whileTrue:
success, img=cap.read()
img=detector.findPose(img)
lmList=detector.findPosition(img,draw=False)
iflen(lmList) !=0:
print(lmList[14])
cv2.circle(img, (lmList[14][1], lmList[14][2]), 10, (0, 0, 255), cv2.FILLED)

cTime=time.time()
fps=2.5/ (cTime-pTime)
pTime=cTime

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

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

if__name__=="__main__":
main()
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
importtime
importmediapipeasmp
importcv2


classposeDetector():

def__init__(self, mode=False, upBody=False, smooth=True,
detectionCon=0.5, trackCon=0.5):
self.mode=mode
self.upBody=upBody
self.smooth=smooth
self.detectionCon=detectionCon
self.trackCon=trackCon


self.mpDraw=mp.solutions.drawing_utils
self.mpPose=mp.solutions.pose
self.pose=self.mpPose.Pose(self.mode,self.upBody,self.smooth,
self.detectionCon,self.trackCon)

deffindPose(self, img, draw=True):
imgRGB=cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
self.results=self.pose.process(imgRGB)
ifself.results.pose_landmarks:
ifdraw:
self.mpDraw.draw_landmarks(img, self.results.pose_landmarks,
self.mpPose.POSE_CONNECTIONS)
returnimg

deffindPosition(self, img, draw=True):
lmList= []
ifself.results.pose_landmarks:
forid, lminenumerate(self.results.pose_landmarks.landmark):
h, w, c=img.shape
#print(id, lm)
cx, cy=int(lm.x*w), int(lm.y*h)
lmList.append([id, cx, cy])
ifdraw:
cv2.circle(img, (cx, cy), 5, (255, 0, 0), cv2.FILLED)
returnlmList


defmain():
cap=cv2.VideoCapture('3.mp4')
pTime=0
detector=poseDetector()
whileTrue:
success, img=cap.read()
img=detector.findPose(img)
lmList=detector.findPosition(img,draw=False)
iflen(lmList) !=0:
print(lmList[14])
cv2.circle(img, (lmList[14][1], lmList[14][2]), 10, (0, 0, 255), cv2.FILLED)

cTime=time.time()
fps=2.5/ (cTime-pTime)
pTime=cTime

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

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

if__name__=="__main__":
main()
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
importtime
importmediapipeasmp
importcv2


classposeDetector():

def__init__(self, mode=False, upBody=False, smooth=True,
detectionCon=0.5, trackCon=0.5):
self.mode=mode
self.upBody=upBody
self.smooth=smooth
self.detectionCon=detectionCon
self.trackCon=trackCon


self.mpDraw=mp.solutions.drawing_utils
self.mpPose=mp.solutions.pose
self.pose=self.mpPose.Pose(self.mode,self.upBody,self.smooth,
self.detectionCon,self.trackCon)

deffindPose(self, img, draw=True):
imgRGB=cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
self.results=self.pose.process(imgRGB)
ifself.results.pose_landmarks:
ifdraw:
self.mpDraw.draw_landmarks(img, self.results.pose_landmarks,
self.mpPose.POSE_CONNECTIONS)
returnimg

deffindPosition(self, img, draw=True):
lmList= []
ifself.results.pose_landmarks:
forid, lminenumerate(self.results.pose_landmarks.landmark):
h, w, c=img.shape
#print(id, lm)
cx, cy=int(lm.x*w), int(lm.y*h)
lmList.append([id, cx, cy])
ifdraw:
cv2.circle(img, (cx, cy), 5, (255, 0, 0), cv2.FILLED)
returnlmList


defmain():
cap=cv2.VideoCapture('3.mp4')
pTime=0
detector=poseDetector()
whileTrue:
success, img=cap.read()
img=detector.findPose(img)
lmList=detector.findPosition(img,draw=False)
iflen(lmList) !=0:
print(lmList[14])
cv2.circle(img, (lmList[14][1], lmList[14][2]), 10, (0, 0, 255), cv2.FILLED)

cTime=time.time()
fps=2.5/ (cTime-pTime)
pTime=cTime

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

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

if__name__=="__main__":
main()