Riferimento per ultralytics/solutions/ai_gym.py
Nota
Questo file è disponibile all'indirizzo https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/solutions/ai_gym .py. Se riscontri un problema, contribuisci a risolverlo inviando una Pull Request 🛠️. Grazie 🙏!
ultralytics.solutions.ai_gym.AIGym
Una classe per gestire i passi in palestra delle persone in un flusso video in tempo reale in base alle loro pose.
Codice sorgente 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__()
Inizializza l'AIGym con i valori predefiniti dei parametri Visual e Image.
Codice sorgente 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')
Configura i parametri spessore_linea, salvataggio immagine e visualizzazione immagine di AIGym.
Parametri:
Nome | Tipo | Descrizione | Predefinito |
---|---|---|---|
kpts_to_check |
list
|
3 punti chiave per il conteggio |
richiesto |
line_thickness |
int
|
Spessore delle linee per i riquadri di delimitazione. |
2
|
view_img |
bool
|
visualizzare l'im0 |
False
|
pose_up_angle |
float
|
Angolo per impostare la posizione di posa |
145.0
|
pose_down_angle |
float
|
Angolo per impostare la posizione di posa |
90.0
|
pose_type |
str
|
"pushup", "pullup" o "abworkout". |
'pullup'
|
Codice sorgente in ultralytics/solutions/ai_gym.py
start_counting(im0, results, frame_count)
Funzione utilizzata per contare i passi della palestra.
Parametri:
Nome | Tipo | Descrizione | Predefinito |
---|---|---|---|
im0 |
ndarray
|
Fotogramma corrente del flusso video. |
richiesto |
results |
list
|
Dati di stima della posa |
richiesto |
frame_count |
int
|
memorizza il conteggio dei fotogrammi correnti |
richiesto |