HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Fix solution functions description + integrations/openvino.md cover image update (#8170)

Browse source 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
Muhammad Rizwan Munawar 2024-02-13 15:33:59 +05:00 • committed by GitHub
parent 52604084de
commit 78f7e475fa
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 18 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

12
ultralytics/solutions/ai_gym.py
View file

@ -46,14 +46,15 @@ class AIGym:
pose_type="pullup",
):
"""
Configures the AIGym line_thickness, save image and view image parameters
Configures the AIGym line_thickness, save image and view image parameters.
Args:
kpts_to_check (list): 3 keypoints for counting
line_thickness (int): Line thickness for bounding boxes.
view_img (bool): display the im0
pose_up_angle (float): Angle to set pose position up
pose_down_angle (float): Angle to set pose position down
pose_type: "pushup", "pullup" or "abworkout"
pose_type (str): "pushup", "pullup" or "abworkout"
"""
self.kpts_to_check = kpts_to_check
self.tf = line_thickness
@ -64,11 +65,12 @@ class AIGym:
def start_counting(self, im0, results, frame_count):
"""
Function used to count the gym steps
Function used to count the gym steps.
Args:
im0 (ndarray): Current frame from the video stream.
results: Pose estimation data
frame_count: store current frame count
results (list): Pose estimation data
frame_count (int): store current frame count
"""
self.im0 = im0
if frame_count == 1: