-
Notifications
You must be signed in to change notification settings - Fork 0
/
AITrainer.py
65 lines (55 loc) · 1.89 KB
/
AITrainer.py
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
import cv2.cv2 as cv
import numpy as np
import time
import PoseEstimationModule as pm
cap = cv.VideoCapture('AiTrainer/curls3.mp4')
# For camera
# cap = cv.VideoCapture(0)
detector = pm.PoseEstimator()
count = 0
dir = 0
pTime = 0
bar = 0
while True:
success, img = cap.read()
# img = cv.imread('AiTrainer/test2.jpg')
img = cv.resize(img, (480, 720))
img = detector.findPose(img, draw=False)
lmList = detector.findPosition(img, draw=False)
# print(lmList)
# Magic starts after this if statement
if len(lmList)!= 0:
# Left Arm
angle = detector.findAngle(img, 11, 13, 15)
# Right Arm
# angle = detector.findAngle(img, 12, 14, 16)
per = np.interp(angle, (210, 300), (0,100))
bar = np.interp(angle, (210, 300), (650,100))
print(angle, per)
# Check for the dumbbell curls
color = (0, 100, 0)
if per == 100:
color= (0,255,0)
if dir == 0: # Going up
count += 0.5
dir = 1
if per == 0:
color = (0, 255, 0)
if dir==1:
count+= 0.5
dir = 0
# print(count)
# Printing the bar
cv.rectangle(img, (400, 100), (450, 650), color, 2)
cv.rectangle(img, (400, int(bar)), (450, 650), color, cv.FILLED)
print(per)
cv.putText(img, f'{int(per)}%', (390, 700), cv.FONT_HERSHEY_PLAIN, 2, color, 2)
# Printing counts of the workout
cv.rectangle(img, (20,20),(130,170),(0,0,0), cv.FILLED)
cv.putText(img, f'{int(count)}', (30,150), cv.FONT_HERSHEY_PLAIN, 10,(0,255,0), 5)
cTime = time.time()
fps = 1 / (cTime - pTime)
pTime = cTime
cv.putText(img, f'FPS:{int(fps)}', (50, 700), cv.FONT_HERSHEY_PLAIN, 2, (0, 255, 0), 2)
cv.imshow("Image", img)
cv.waitKey(5)