Reference for ultralytics/solutions/ai_gym.py
Note
This file is available at https://github.com/ultralytics/ultralytics/blob/main/ultralytics/solutions/ai_gym.py. If you spot a problem please help fix it by contributing a Pull Request 🛠️. Thank you 🙏!
ultralytics.solutions.ai_gym.AIGym
A class to manage the gym steps of people in a real-time video stream based on their poses.
Source code in ultralytics/solutions/ai_gym.py
9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 |
|
__init__()
Initializes the AIGym with default values for Visual and Image parameters.
Source code in ultralytics/solutions/ai_gym.py
set_args(kpts_to_check, line_thickness=2, view_img=False, pose_up_angle=145.0, pose_down_angle=90.0, pose_type='pullup')
Configures the AIGym line_thickness, save image and view image parameters.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
kpts_to_check |
list
|
3 keypoints for counting |
required |
line_thickness |
int
|
Line thickness for bounding boxes. |
2
|
view_img |
bool
|
display the im0 |
False
|
pose_up_angle |
float
|
Angle to set pose position up |
145.0
|
pose_down_angle |
float
|
Angle to set pose position down |
90.0
|
pose_type |
str
|
"pushup", "pullup" or "abworkout" |
'pullup'
|
Source code in ultralytics/solutions/ai_gym.py
start_counting(im0, results, frame_count)
Function used to count the gym steps.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
im0 |
ndarray
|
Current frame from the video stream. |
required |
results |
list
|
Pose estimation data |
required |
frame_count |
int
|
store current frame count |
required |