Update poseDetector.py

Freedisch · Feb 12, 2024 · 1028ce4 · 1028ce4
1 parent d076619
commit 1028ce4
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/aimodel/trainedmodel/poseDetector.py b/aimodel/trainedmodel/poseDetector.py
@@ -14,7 +14,6 @@ def __init__(self, mode=False, upBody=False, smooth=True, detectionCon=0.5, trac
 
         self.mpPose = mp.solutions.pose
         self.mpDraw = mp.solutions.drawing_utils
-        #self.pose = self.mpPose.Pose(self.mode, self.upBody, self.smooth, self.detectionCon, self.trackCon)
         self.pose = self.mpPose.Pose(self.mode, 1, True, self.upBody, self.smooth, self.detectionCon, self.trackCon)
 
     def findPose(self, img, draw=True):
@@ -24,11 +23,6 @@ def findPose(self, img, draw=True):
             if draw:
                 self.mpDraw.draw_landmarks(img, results.pose_landmarks, self.mpPose.POSE_CONNECTIONS)
         return img
-    # for id, lm in enumerate(results.pose_landmarks.landmark):
-    #     h, w, c = img.shape
-    #     cx, cy = int(lm.x * w), int(lm.y * h)
-    #     cv2.circle(img, (cx, cy), 5, (255, 0, 0), cv2.FILLED)
-
 
 def main():
     cap = cv2.VideoCapture(0)
@@ -49,4 +43,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()