From fd4faa45b27287c51929f101d9258cd90de00911 Mon Sep 17 00:00:00 2001
From: Victor Mylle <victor.mylle@hotmail.com>
Date: Fri, 14 Apr 2023 14:27:51 +0200
Subject: [PATCH] Little fix

---
 webcam.py | 68 +++++++++++++++++++++++++++----------------------------
 1 file changed, 34 insertions(+), 34 deletions(-)

diff --git a/webcam.py b/webcam.py
index ef37c56..f437483 100644
--- a/webcam.py
+++ b/webcam.py
@@ -298,45 +298,45 @@ def make_prediction(keypoints):
     return c, score
     
 # 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():
     # read frame
     ret, frame = cap.read()
-    if frame is None:
-        break
     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():
+#     # 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)
\ No newline at end of file
+# label, score = make_prediction(buffer)
+# print(label, score)
\ No newline at end of file