Référence pour ultralytics/solutions/distance_calculation.py
Note
Ce fichier est disponible à l'adresse https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/solutions/distance_calculation .py. Si tu repères un problème, aide à le corriger en contribuant à une Pull Request 🛠️. Merci 🙏 !
ultralytics.solutions.distance_calculation.DistanceCalculation
Une classe pour calculer la distance entre deux objets dans un flux vidéo en temps réel en fonction de leurs traces.
Code source dans ultralytics/solutions/distance_calculation.py
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 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 |
|
__init__()
Initialise la classe de calcul de distance avec des valeurs par défaut pour les paramètres Visuel, Image, piste et distance. et de la distance.
Code source dans ultralytics/solutions/distance_calculation.py
calculate_centroid(box)
Calcule le centroïde de la boîte de délimitation.
Paramètres :
Nom | Type | Description | Défaut |
---|---|---|---|
box |
list
|
Données de la boîte de délimitation |
requis |
calculate_distance(centroid1, centroid2)
Calcule la distance entre deux centroïdes.
Paramètres :
Nom | Type | Description | Défaut |
---|---|---|---|
centroid1 |
point
|
Premières données de la boîte de délimitation |
requis |
centroid2 |
point
|
Données de la deuxième boîte de délimitation |
requis |
Code source dans ultralytics/solutions/distance_calculation.py
display_frames()
Cadre d'affichage.
Code source dans ultralytics/solutions/distance_calculation.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/distance_calculation.py
mouse_event_for_distance(event, x, y, flags, param)
Cette fonction est conçue pour déplacer une région avec les événements de la souris dans un flux vidéo en temps réel.
Paramètres :
Nom | Type | Description | Défaut |
---|---|---|---|
event |
int
|
Le type d'événement de la souris (par exemple, cv2.EVENT_MOUSEMOVE, cv2.EVENT_LBUTTONDOWN, etc.) |
requis |
x |
int
|
La coordonnée x du pointeur de la souris. |
requis |
y |
int
|
La coordonnée y du pointeur de la souris. |
requis |
flags |
int
|
Tous les drapeaux associés à l'événement (par exemple, cv2.EVENT_FLAG_CTRLKEY, cv2.EVENT_FLAG_SHIFTKEY, etc.) |
requis |
param |
dict
|
Paramètres supplémentaires que tu peux vouloir passer à la fonction. |
requis |
Code source dans ultralytics/solutions/distance_calculation.py
set_args(names, pixels_per_meter=10, view_img=False, line_thickness=2, line_color=(255, 255, 0), centroid_color=(255, 0, 255))
Configure les paramètres de calcul et d'affichage de la distance.
Paramètres :
Nom | Type | Description | Défaut |
---|---|---|---|
names |
dict
|
détection d'objets classes noms |
requis |
pixels_per_meter |
int
|
Nombre de pixels par mètre |
10
|
view_img |
bool
|
Drapeau indiquant l'affichage du cadre |
False
|
line_thickness |
int
|
Épaisseur du trait pour les boîtes de délimitation. |
2
|
line_color |
RGB
|
couleur de la ligne des centroïdes |
(255, 255, 0)
|
centroid_color |
RGB
|
couleurs des centroïdes des boîtes |
(255, 0, 255)
|
Code source dans ultralytics/solutions/distance_calculation.py
start_process(im0, tracks)
Calcule la distance entre deux boîtes de délimitation 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 |
Code source dans ultralytics/solutions/distance_calculation.py
Créé le 2024-01-05, Mis à jour le 2024-05-08
Auteurs : Burhan-Q (1), AyushExel (1), RizwanMunawar (1)