Update aiModel.py

Freedisch · Feb 12, 2024 · 638b2d4 · 638b2d4
1 parent f2bc7e3
commit 638b2d4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/aimodel/trainedmodel/aiModel.py b/aimodel/trainedmodel/aiModel.py
@@ -10,18 +10,18 @@
 while True:
     success, img = cap.read()
     img = cv2.resize(img, (1280, 720))
-    # img = cv2.imread("AiTrainer/test.jpg")
+
     img = detector.findPose(img, False)
     lmList = detector.findPosition(img, False)
     # print(lmList)
     if len(lmList) != 0:
         # Right Arm
         angle = detector.findAngle(img, 12, 14, 16)
+
         # # Left Arm
-        #angle = detector.findAngle(img, 11, 13, 15,False)
         per = np.interp(angle, (210, 310), (0, 100))
         bar = np.interp(angle, (220, 310), (650, 100))
-        # print(angle, per)
+
         # Check for the dumbbell curls
         color = (255, 0, 255)
         if per == 100:
@@ -50,4 +50,4 @@
     cv2.putText(img, str(int(fps)), (50, 100), cv2.FONT_HERSHEY_PLAIN, 5,
                 (255, 0, 0), 5)
     cv2.imshow("Image", img)
-    cv2.waitKey(1)
+    cv2.waitKey(1)