diff --git a/aimodel/trainedmodel/aiModel.py b/aimodel/trainedmodel/aiModel.py index 95344a4..f86e2a0 100644 --- 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) \ No newline at end of file + cv2.waitKey(1)