Amazing-Python-Scripts

1
import cv2
2
import mediapipe as mp
3
import time
4

5
mpDraw = mp.solutions.drawing_utils
6
mpPose = mp.solutions.pose
7
pose = mpPose.Pose()
8

9
cap = cv2.VideoCapture(0)
10
# cap = cv2.VideoCapture('/path/to/video/name_of_video.mp4')
11
pTime = 0
12

13
while True:
14
    success, img = cap.read()
15
    imgRGB = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
16
    results = pose.process(imgRGB)
17
    # print(results.pose_landmarks)
18
    if results.pose_landmarks:
19
        mpDraw.draw_landmarks(img, results.pose_landmarks,
20
                              mpPose.POSE_CONNECTIONS)
21
        for id, lm in enumerate(results.pose_landmarks.landmark):
22
            h, w, c = img.shape
23
            print(id, lm)
24
            cx, cy = int(lm.x*w), int(lm.y*h)
25
            cv2.circle(img, (cx, cy), 5, (255, 255, 0), cv2.FILLED)
26

27
    cTime = time.time()
28
    fps = 1/(cTime - pTime)
29
    pTime = cTime
30

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

34
    cv2.imshow("Image", img)
35
    cv2.waitKey(1)
36

37
cap.release()
38
cv2.destroyAllWindow
39