Référence pour ultralytics/solutions/speed_estimation.py
Note
Ce fichier est disponible à l'adresse https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/solutions/speed_estimation .py. Si tu repères un problème, aide à le corriger en contribuant à une Pull Request 🛠️. Merci 🙏 !
ultralytics.solutions.speed_estimation.SpeedEstimator
Une classe pour estimer la vitesse des objets dans un flux vidéo en temps réel en fonction de leurs traces.
Code source dans 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__()
Initialise la classe de l'estimateur de vitesse avec des valeurs par défaut pour les paramètres Visuel, Image, piste et vitesse.
Code source dans ultralytics/solutions/speed_estimation.py
calculate_speed(trk_id, track)
Calcul de la vitesse de l'objet.
Paramètres :
Nom | Type | Description | DĂ©faut |
---|---|---|---|
trk_id |
int
|
l'identifiant de la piste de l'objet. |
requis |
track |
list
|
historique de suivi des traces dessin du chemin |
requis |
Code source dans ultralytics/solutions/speed_estimation.py
display_frames()
estimate_speed(im0, tracks, region_color=(255, 0, 0))
Calcule l'objet en fonction des données de suivi.
Paramètres :
Nom | Type | Description | DĂ©faut |
---|---|---|---|
im0 |
nd array
|
Image |
requis |
tracks |
list
|
Liste des pistes obtenues lors du processus de suivi des objets. |
requis |
region_color |
tuple
|
Couleur à utiliser pour dessiner les régions. |
(255, 0, 0)
|
Code source dans ultralytics/solutions/speed_estimation.py
extract_tracks(tracks)
Extrait les résultats des données fournies.
Paramètres :
Nom | Type | Description | DĂ©faut |
---|---|---|---|
tracks |
list
|
Liste des pistes obtenues lors du processus de suivi des objets. |
requis |
Code source dans ultralytics/solutions/speed_estimation.py
plot_box_and_track(track_id, box, cls, track)
Trace la piste et la boîte de délimitation.
Paramètres :
Nom | Type | Description | DĂ©faut |
---|---|---|---|
track_id |
int
|
l'identifiant de la piste de l'objet. |
requis |
box |
list
|
données du cadre de délimitation de l'objet |
requis |
cls |
str
|
nom de la classe d'objets |
requis |
track |
list
|
historique de suivi des traces dessin du chemin |
requis |
Code source dans ultralytics/solutions/speed_estimation.py
set_args(reg_pts, names, view_img=False, line_thickness=2, region_thickness=5, spdl_dist_thresh=10)
Configure les paramètres d'estimation et d'affichage de la vitesse.
Paramètres :
Nom | Type | Description | DĂ©faut |
---|---|---|---|
reg_pts |
list
|
Liste initiale de points définissant la région de calcul de la vitesse. |
requis |
names |
dict
|
détection d'objets classes noms |
requis |
view_img |
bool
|
Drapeau indiquant l'affichage du cadre |
False
|
line_thickness |
int
|
Épaisseur du trait pour les boîtes de délimitation. |
2
|
region_thickness |
int
|
Epaisseur de la région d'estimation de la vitesse |
5
|
spdl_dist_thresh |
int
|
Seuil de distance euclidienne pour la ligne de vitesse |
10
|
Code source dans ultralytics/solutions/speed_estimation.py
store_track_info(track_id, box)
Enregistre les données de la piste.
Paramètres :
Nom | Type | Description | DĂ©faut |
---|---|---|---|
track_id |
int
|
l'identifiant de la piste de l'objet. |
requis |
box |
list
|
données du cadre de délimitation de l'objet |
requis |
Code source dans ultralytics/solutions/speed_estimation.py
Créé le 2024-01-05, Mis à jour le 2024-05-08
Auteurs : Burhan-Q (1), AyushExel (1), RizwanMunawar (1)