mainBensVersion.py

import cv2
##import serial
import time
import mediapipe as mp

mpDraw = mp.solutions.drawing_utils
mpPose = mp.solutions.pose
pose = mpPose.Pose()

cap = cv2.VideoCapture(0)
pTime = 0
# 2d array to track the position of the landmarks 50x50
arr = [['.' for i in range(200)] for j in range(70)]
k=0
while k<100:
    success, img = cap.read()
    imgRGB = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
    results = pose.process(imgRGB)
    # print(results.pose_landmarks)
    if results.pose_landmarks:
        mpDraw.draw_landmarks(img, results.pose_landmarks, mpPose.POSE_CONNECTIONS)
        for id, lm in enumerate(results.pose_landmarks.landmark):
            h, w, c = img.shape
            ##print(id, lm)
            cx, cy = int(lm.xw), int(lm.yh)
            cv2.circle(img, (cx, cy), 5, (255, 0, 0), cv2.FILLED)
            ##print(lm.yh,lm.xw)
            arr[int(lm.yh/40)][int(lm.xw/15)] = 0
            ##print(h, w)

    cTime = time.time()
    fps = 1/(cTime-pTime)
    pTime = cTime
    k+=1

    cv2.putText(img, str(int(fps)), (70, 50), cv2.FONT_HERSHEY_PLAIN, 3, (255, 0, 255), 3)
    cv2.imshow("Image", img)
    cv2.waitKey(1)

# print array in grid format
for i in range(25):
    for j in range(40):
        print(arr[i][j], end="")
    print()


# 12 zones, 3upx4wide grid
# 16 pieces per zone, 4x4 grid
# 192 pieces total

# 12 pieces upx16 pieces wide