25 fps
#hand.py
import os
import cv2
import mediapipe as mp
import time
cap = cv2.VideoCapture("assets/sign_language.mp4")
# Define the codec and create VideoWriter object
fps = 25.175
frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
size = (int(frame_width), int(frame_height))
fourcc = cv2.VideoWriter_fourcc('m', 'p', '4', 'v')
path = 'C:/Users/zchen/PycharmProjects/opencv/googleNet/record'
out = cv2.VideoWriter()
success = out.open(os.path.join(path, "hand tracking mediapipe.mov"), fourcc, fps, size, True)
frame_count = 0
t1 = time.time()
mpHands = mp.solutions.hands
hands = mpHands.Hands()
mpDraw = mp.solutions.drawing_utils
while True:
ret, frame = cap.read()
if ret == False: break
frame_RGB = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
results = hands.process(frame_RGB)
if results.multi_hand_landmarks:
for handLms in results.multi_hand_landmarks:
for id, lm in enumerate(handLms.landmark):
#print(id, lm)
h, w, c = frame.shape
cx, cy = int(lm.x * w), int(lm.y * h)
#print(id, cx, cy)
#draw circle at thumb tip
if id == 4:
cv2.circle(frame, (cx, cy), 15, (255, 0, 255), cv2.FILLED)
mpDraw.draw_landmarks(frame, handLms, mpHands.HAND_CONNECTIONS)
cv2.imshow("frame", frame)
out.write(frame)
frame_count += 1
if cv2.waitKey(1) == ord('q'):
break
if cv2.waitKey(1) == ord('p'):
cv2.waitKey(-1) # wait until any key is pressed
t2 = time.time()
# calculate FPS
fps = str(float(frame_count / float(t2 - t1))) + ' FPS'
print("Frames processed: {}".format(frame_count))
print("Elapsed time: {:.2f}".format(float(t2 - t1)))
print("FPS: {}".format(fps))
cap.release()
cv2.destroyAllWindows()
#reference:
No comments:
Post a Comment