Computer-Vision-Python/PoseModule.py at main · funinkina/Computer-Vision-Python · 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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
importmediapipeasmp
importtime
importmath
importcv2


classposeDetector():
def__init__(self, mode=False, upBody=False, modelComplexity=1, smooth=True,
detectionCon=0.5, trackCon=0.5):
self.mode=mode
self.upBody=upBody
self.modelComplex=modelComplexity
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.modelComplex, 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):
self.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)
self.lmList.append([id, cx, cy])
ifdraw:
cv2.circle(img, (cx, cy), 5, (255, 0, 0), cv2.FILLED)
returnself.lmList


deffindAngle(self, img, p1, p2, p3, draw=True):
# Get the landmarks
x1, y1=self.lmList[p1][1:]
x2, y2=self.lmList[p2][1:]
x3, y3=self.lmList[p3][1:]
# Calculate the Angle
angle=math.degrees(math.atan2(y3-y2, x3-x2) -
math.atan2(y1-y2, x1-x2))
ifangle<0:
angle+=360
# print(angle)
# Draw
ifdraw:
cv2.line(img, (x1, y1), (x2, y2), (255, 255, 255), 3)
cv2.line(img, (x3, y3), (x2, y2), (255, 255, 255), 3)
cv2.circle(img, (x1, y1), 10, (0, 0, 255), cv2.FILLED)
cv2.circle(img, (x1, y1), 15, (0, 0, 255), 2)
cv2.circle(img, (x2, y2), 10, (0, 0, 255), cv2.FILLED)
cv2.circle(img, (x2, y2), 15, (0, 0, 255), 2)
cv2.circle(img, (x3, y3), 10, (0, 0, 255), cv2.FILLED)
cv2.circle(img, (x3, y3), 15, (0, 0, 255), 2)
cv2.putText(img, str(int(angle)), (x2-50, y2+50),
cv2.FONT_HERSHEY_PLAIN, 2, (0, 0, 255), 2)
returnangle


defmain():
cap=cv2.VideoCapture('Poses/dancing.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]), 15, (0, 0, 255), cv2.FILLED)
cTime=time.time()
fps=1/ (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()