Pushups_Counter/main.py at main · Manish7272/Pushups_Counter · 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
import numpy as np
import cv2
import time

import cvzone
from cvzone.PoseModule import PoseDetector

cap = cv2.VideoCapture(0)

ctime = 0
ptime = 0
direction = 0
push_ups = 0

# --------------------------- rgb(37, 150, 190)
b_color = (0, 0, 0)
blue_color = (190, 150, 37)


detector = PoseDetector()
while True:
    _, img = cap.read()
    img = detector.findPose(img)
    lmlist, bbox = detector.findPosition(img, draw=False)
    if lmlist:

        a1 = detector.findAngle(img, 12, 14, 16)
        a2 = detector.findAngle(img, 15, 13, 11)

        per_val1 = int(np.interp(a1, (90, 170), (100, 0)))
        per_val2 = int(np.interp(a2, (90, 170), (100, 0)))

        bar_val1 = int(np.interp(per_val1, (0, 100), (40+350, 40)))
        bar_val2 = int(np.interp(per_val2, (0, 100), (40+350, 40)))

        # 1st bar
        cv2.rectangle(img, (570, bar_val1), (570 + 35, 40 + 350), (0, 0, 255), cv2.FILLED)
        cv2.rectangle(img, (570, 40), (570 + 35, 40+350), (), 2)

        # 2st bar
        cv2.rectangle(img, (35, bar_val2), (35 + 35, 40 + 350), (0, 0, 255), cv2.FILLED)
        cv2.rectangle(img, (35, 40), (35 + 35, 40 + 350), (), 2)

        # bar %
        cvzone.putTextRect(img, f"{per_val1} %", (570, 25), 1.3, 2, colorT=blue_color, colorR=(255, 255, 255), border=1, colorB=b_color)  # (B,G,R)
        cvzone.putTextRect(img, f"{per_val2} %", (25, 25), 1.3, 2, colorT=blue_color, colorR=(255, 255, 255), border=1, colorB=b_color)  # (B,G,R)

        if per_val1 == 100 and per_val2 == 100:
            if direction == 0:
                push_ups += 0.5
                direction = 1
                color = (0, 255, 0)

        elif per_val1 == 0 and per_val2 == 0:
            if direction == 1:
                push_ups += 0.5
                direction = 0
                color = (0, 255, 0)
        else:
            color = (0, 0, 255)

        cvzone.putTextRect(img, f"Push_ups : {int(push_ups)}", (200, 35), 2, 2, colorT=(0, 0, 255), colorR=(0, 255, 0), colorB=(), border=1)
        cvzone.putTextRect(img, "Left Hand", (15, 350+100), 1.5, 2, colorT=(255, 255, 255), colorR=blue_color, colorB=b_color, border=1)
        cvzone.putTextRect(img, "Right Hand", (485, 350+100), 1.5, 2, colorT=(255, 255, 255), colorR=blue_color, colorB=b_color, border=1)

        print(push_ups)

    ctime = time.time()
    fps = 1/(ctime-ptime)
    ptime = ctime
    # cvzone.putTextRect(img, f"FPS :{int(fps)}", (288, 440), 1.5, 1, colorT=(255, 255, 255), colorR=(1, 155, 0), border=2,colorB=(0, 0, 0))

    cv2.imshow("Posh-ups Counter", img)
    if cv2.waitKey(1) == ord("q"):
        break