WeSign/spoterembedding
1
0
Fork 0
Code Issues Pull Requests 1 Packages Projects Releases Wiki Activity

Little fix

Browse Source
This commit is contained in:
Victor Mylle
2023-04-14 14:27:51 +02:00
parent 8771fe0721
commit fd4faa45b2
1 changed files with 34 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

68
webcam.py
View File

@@ -298,45 +298,45 @@ def make_prediction(keypoints):
return c, score return c, score
# open webcam stream # open webcam stream
# cap = cv2.VideoCapture(0) cap = cv2.VideoCapture(0)
# while cap.isOpened():
# # read frame
# ret, frame = cap.read()
# pose = extract_keypoints(frame)
# if pose is None:
# cv2.imshow('MediaPipe Hands', frame)
# continue
# buffer.append(pose)
# if len(buffer) > 15:
# buffer.pop(0)
# if len(buffer) == 15:
# label, score = make_prediction(buffer)
# # draw label
# cv2.putText(frame, label, (10, 30), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 255), 2, cv2.LINE_AA)
# cv2.putText(frame, str(score), (10, 60), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 255), 2, cv2.LINE_AA)
# # Show the frame
# cv2.imshow('MediaPipe Hands', frame)
# # Wait for key press to exit
# if cv2.waitKey(5) & 0xFF == 27:
# break
# open video A.mp4
cap = cv2.VideoCapture('E.mp4')
while cap.isOpened(): while cap.isOpened():
# read frame # read frame
ret, frame = cap.read() ret, frame = cap.read()
if frame is None:
break
pose = extract_keypoints(frame) pose = extract_keypoints(frame)
buffer.append(pose) if pose is None:
cv2.imshow('MediaPipe Hands', frame)
continue
buffer.append(pose)
if len(buffer) > 15:
buffer.pop(0)
if len(buffer) == 15:
label, score = make_prediction(buffer) label, score = make_prediction(buffer)
print(label, score)
# draw label
cv2.putText(frame, label, (10, 30), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 255), 2, cv2.LINE_AA)
cv2.putText(frame, str(score), (10, 60), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 255), 2, cv2.LINE_AA)
# Show the frame
cv2.imshow('MediaPipe Hands', frame)
# Wait for key press to exit
if cv2.waitKey(5) & 0xFF == 27:
break
# open video A.mp4
# cap = cv2.VideoCapture('E.mp4')
# while cap.isOpened():
# # read frame
# ret, frame = cap.read()
# if frame is None:
# break
# pose = extract_keypoints(frame)
# buffer.append(pose)
# label, score = make_prediction(buffer)
# print(label, score)