Referencia para ultralytics/solutions/ai_gym.py
Nota
Este archivo está disponible en https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/solutions/ai_gym .py. Si detectas algún problema, por favor, ayuda a solucionarlo contribuyendo con una Pull Request 🛠️. ¡Gracias 🙏!
ultralytics.solutions.ai_gym.AIGym
Una clase para gestionar los pasos de gimnasia de las personas en un flujo de vídeo en tiempo real basado en sus posturas.
Código fuente en 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__()
Inicializa el AIGym con valores por defecto para los parámetros Visual e Imagen.
Código fuente en 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')
Configura los parámetros grosor_línea, guardar imagen y ver imagen de AIGym.
Parámetros:
Nombre | Tipo | Descripción | Por defecto |
---|---|---|---|
kpts_to_check |
list
|
3 puntos clave para el recuento |
necesario |
line_thickness |
int
|
Grosor de línea de los cuadros delimitadores. |
2
|
view_img |
bool
|
mostrar la im0 |
False
|
pose_up_angle |
float
|
Ángulo para fijar la postura |
145.0
|
pose_down_angle |
float
|
Ángulo para fijar la postura |
90.0
|
pose_type |
str
|
"flexiones", "tracciones" o "ejercicios abdominales" |
'pullup'
|
Código fuente en ultralytics/solutions/ai_gym.py
start_counting(im0, results, frame_count)
Función utilizada para contar los pasos del gimnasio.
Parámetros:
Nombre | Tipo | Descripción | Por defecto |
---|---|---|---|
im0 |
ndarray
|
Fotograma actual del flujo de vídeo. |
necesario |
results |
list
|
Datos de estimación de la pose |
necesario |
frame_count |
int
|
almacenar el recuento de fotogramas actual |
necesario |