Referência para ultralytics/solutions/ai_gym.py
Nota
Este ficheiro está disponível em https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/solutions/ai_gym.py. Se detectares um problema, por favor ajuda a corrigi-lo contribuindo com um Pull Request 🛠️. Obrigado 🙏!
ultralytics.solutions.ai_gym.AIGym
Uma classe para gerir os passos de ginástica das pessoas num fluxo de vídeo em tempo real com base nas suas poses.
Código fonte em 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 o AIGym com os valores por defeito dos parâmetros Visual e Imagem.
Código fonte em 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 os parâmetros AIGym line_thickness, save image e view image.
Parâmetros:
Nome | Tipo | Descrição | Predefinição |
---|---|---|---|
kpts_to_check |
list
|
3 pontos-chave para a contagem |
necessário |
line_thickness |
int
|
Espessura da linha para caixas delimitadoras. |
2
|
view_img |
bool
|
apresenta o im0 |
False
|
pose_up_angle |
float
|
Ângulo para definir a posição da pose |
145.0
|
pose_down_angle |
float
|
Ângulo para definir a posição da pose |
90.0
|
pose_type |
str
|
"pushup", "pullup" ou "abworkout" |
'pullup'
|
Código fonte em ultralytics/solutions/ai_gym.py
start_counting(im0, results, frame_count)
Função utilizada para contar os passos do ginásio.
Parâmetros:
Nome | Tipo | Descrição | Predefinição |
---|---|---|---|
im0 |
ndarray
|
Fotograma atual do fluxo de vídeo. |
necessário |
results |
list
|
Dados de estimativa de pose |
necessário |
frame_count |
int
|
guarda a contagem atual de fotogramas |
necessário |