Riferimento per ultralytics/solutions/speed_estimation.py
Nota
Questo file è disponibile all'indirizzo https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/solutions/speed_estimation .py. Se riscontri un problema, contribuisci a risolverlo inviando una Pull Request 🛠️. Grazie 🙏!
ultralytics.solutions.speed_estimation.SpeedEstimator
Una classe per stimare la velocità degli oggetti in un flusso video in tempo reale in base alle loro tracce.
Codice sorgente in ultralytics/solutions/speed_estimation.py
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 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 |
|
__init__()
Inizializza la classe speed-estimator con i valori predefiniti dei parametri Visual, Image, track e speed.
Codice sorgente in ultralytics/solutions/speed_estimation.py
calculate_speed(trk_id, track)
Calcolo della velocità dell'oggetto.
Parametri:
Nome | Tipo | Descrizione | Predefinito |
---|---|---|---|
trk_id |
int
|
oggetto traccia id. |
richiesto |
track |
list
|
cronologia di tracciamento per il disegno del percorso dei binari |
richiesto |
Codice sorgente in ultralytics/solutions/speed_estimation.py
display_frames()
estimate_speed(im0, tracks, region_color=(255, 0, 0))
Calcola l'oggetto in base ai dati di tracciamento.
Parametri:
Nome | Tipo | Descrizione | Predefinito |
---|---|---|---|
im0 |
nd array
|
Immagine |
richiesto |
tracks |
list
|
Elenco delle tracce ottenute dal processo di tracciamento dell'oggetto. |
richiesto |
region_color |
tuple
|
Colore da utilizzare per disegnare le regioni. |
(255, 0, 0)
|
Codice sorgente in ultralytics/solutions/speed_estimation.py
extract_tracks(tracks)
Estrae i risultati dai dati forniti.
Parametri:
Nome | Tipo | Descrizione | Predefinito |
---|---|---|---|
tracks |
list
|
Elenco delle tracce ottenute dal processo di tracciamento dell'oggetto. |
richiesto |
Codice sorgente in ultralytics/solutions/speed_estimation.py
plot_box_and_track(track_id, box, cls, track)
Traccia e riquadro di delimitazione.
Parametri:
Nome | Tipo | Descrizione | Predefinito |
---|---|---|---|
track_id |
int
|
oggetto traccia id. |
richiesto |
box |
list
|
dati del rettangolo di selezione dell'oggetto |
richiesto |
cls |
str
|
nome della classe dell'oggetto |
richiesto |
track |
list
|
cronologia di tracciamento per il disegno del percorso dei binari |
richiesto |
Codice sorgente in ultralytics/solutions/speed_estimation.py
set_args(reg_pts, names, view_img=False, line_thickness=2, region_thickness=5, spdl_dist_thresh=10)
Configura i parametri di stima e visualizzazione della velocità.
Parametri:
Nome | Tipo | Descrizione | Predefinito |
---|---|---|---|
reg_pts |
list
|
Elenco iniziale di punti che definiscono la regione di calcolo della velocità. |
richiesto |
names |
dict
|
nomi delle classi di rilevamento degli oggetti |
richiesto |
view_img |
bool
|
Flag che indica la visualizzazione della cornice |
False
|
line_thickness |
int
|
Spessore delle linee per i riquadri di delimitazione. |
2
|
region_thickness |
int
|
Spessore della regione di stima della velocità |
5
|
spdl_dist_thresh |
int
|
Soglia di distanza euclidea per la linea di velocità |
10
|
Codice sorgente in ultralytics/solutions/speed_estimation.py
store_track_info(track_id, box)
Memorizza i dati della traccia.
Parametri:
Nome | Tipo | Descrizione | Predefinito |
---|---|---|---|
track_id |
int
|
oggetto traccia id. |
richiesto |
box |
list
|
dati del rettangolo di selezione dell'oggetto |
richiesto |