Calcul de distance avec Ultralytics YOLO26
Qu'est-ce que le calcul de distance ?
Mesurer l'écart entre deux objets est connu sous le nom de calcul de distance au sein d'un espace spécifié. Dans le cas d'Ultralytics YOLO26, le centroïde de la bounding box est utilisé pour calculer la distance pour les bounding boxes mises en évidence par l'utilisateur.
Watch: How to estimate distance between detected objects with Ultralytics YOLO in Pixels 🚀
Visuels
| Calcul de distance avec Ultralytics YOLO26 |
|---|
![]() |
Avantages du calcul de distance
- Précision de localisation : Améliore le positionnement spatial précis dans les tâches de vision par ordinateur.
- Estimation de taille : Permet d'estimer la taille d'un objet pour une meilleure compréhension contextuelle.
- Compréhension de la scène : Améliore la compréhension des scènes 3D pour une meilleure prise de décision dans des applications telles que les véhicules autonomes et les systèmes de surveillance.
- Évitement de collision : Permet aux systèmes de détecter les collisions potentielles en surveillant les distances entre les objets en mouvement.
- Analyse spatiale : Facilite l'analyse des relations et des interactions entre les objets au sein de l'environnement surveillé.
Calcul de distance
- Clique sur deux bounding boxes avec le bouton gauche de la souris pour calculer la distance.
- Utilise le bouton droit de la souris pour supprimer tous les points tracés.
- Fais un clic gauche n'importe où dans l'image pour ajouter de nouveaux points.
La distance est une estimation
La distance est une estimation et peut ne pas être totalement précise car elle est calculée à l'aide de données 2D, qui manquent d'informations de profondeur.
import cv2
from ultralytics import solutions
cap = cv2.VideoCapture("path/to/video.mp4")
assert cap.isOpened(), "Error reading video file"
# Video writer
w, h, fps = (int(cap.get(x)) for x in (cv2.CAP_PROP_FRAME_WIDTH, cv2.CAP_PROP_FRAME_HEIGHT, cv2.CAP_PROP_FPS))
video_writer = cv2.VideoWriter("distance_output.avi", cv2.VideoWriter_fourcc(*"mp4v"), fps, (w, h))
# Initialize distance calculation object
distancecalculator = solutions.DistanceCalculation(
model="yolo26n.pt", # path to the YOLO26 model file.
show=True, # display the output
)
# Process video
while cap.isOpened():
success, im0 = cap.read()
if not success:
print("Video frame is empty or processing is complete.")
break
results = distancecalculator(im0)
print(results) # access the output
video_writer.write(results.plot_im) # write the processed frame.
cap.release()
video_writer.release()
cv2.destroyAllWindows() # destroy all opened windowsArguments de DistanceCalculation()
Voici un tableau avec les arguments de DistanceCalculation :
| Argument | Type | Défaut | Description |
|---|---|---|---|
model | str | None | Chemin d'accès vers un fichier de modèle Ultralytics YOLO. |
Tu peux également utiliser divers arguments de track dans la solution DistanceCalculation.
| Argument | Type | Défaut | Description |
|---|---|---|---|
tracker | str | 'botsort.yaml' | Spécifie l'algorithme de suivi à utiliser, par exemple bytetrack.yaml ou botsort.yaml. |
conf | float | 0.1 | Définit le seuil de confiance pour les détections ; des valeurs plus basses permettent de suivre davantage d'objets mais peuvent inclure des faux positifs. |
iou | float | 0.7 | Définit le seuil Intersection over Union (IoU) pour filtrer les détections qui se chevauchent. |
classes | list | None | Filtre les résultats par index de classe. Par exemple, classes=[0, 2, 3] suit uniquement les classes spécifiées. |
verbose | bool | True | Contrôle l'affichage des résultats de suivi, fournissant une sortie visuelle des objets suivis. |
device | str | None | Spécifie le périphérique pour l'inférence (par exemple, cpu, cuda:0 ou 0). Permet aux utilisateurs de choisir entre le CPU, un GPU spécifique ou d'autres dispositifs de calcul pour l'exécution du modèle. |
De plus, les arguments de visualisation suivants sont disponibles :
| Argument | Type | Défaut | Description |
|---|---|---|---|
show | bool | False | Si True, affiche les images ou vidéos annotées dans une fenêtre. Utile pour un retour visuel immédiat lors du développement ou des tests. |
line_width | int or None | None | Spécifie l'épaisseur de ligne des boîtes englobantes. Si None, l'épaisseur de ligne est automatiquement ajustée en fonction de la taille de l'image. Fournit une personnalisation visuelle pour plus de clarté. |
show_conf | bool | True | Affiche le score de confiance pour chaque détection à côté de l'étiquette. Donne un aperçu de la certitude du modèle pour chaque détection. |
show_labels | bool | True | Affiche les étiquettes pour chaque détection dans la sortie visuelle. Fournit une compréhension immédiate des objets détectés. |
Détails de mise en œuvre
La classe DistanceCalculation fonctionne en suivant les objets à travers les images vidéo et en calculant la distance euclidienne entre les centroïdes des bounding boxes sélectionnées. Lorsque tu cliques sur deux objets, la solution :
- Extrait les centroïdes (points centraux) des bounding boxes sélectionnées
- Calcule la distance euclidienne entre ces centroïdes en pixels
- Affiche la distance sur l'image avec une ligne reliant les objets
La mise en œuvre utilise la méthode mouse_event_for_distance pour gérer les interactions avec la souris, permettant aux utilisateurs de sélectionner des objets et d'effacer les sélections si nécessaire. La méthode process gère le traitement image par image, le suivi des objets et le calcul des distances.
Applications
Le calcul de distance avec YOLO26 a de nombreuses applications pratiques :
- Analyse de détail : Mesure la proximité des clients avec les produits et analyse l'efficacité de l'agencement du magasin
- Sécurité industrielle : Surveille les distances de sécurité entre les travailleurs et les machines
- Gestion du trafic : Analyse l'espacement des véhicules et détecte le talonnage
- Analyse sportive : Calcule les distances entre les joueurs, le ballon et les positions clés sur le terrain
- Santé : Assure une distanciation appropriée dans les zones d'attente et surveille les mouvements des patients
- Robotique : Permet aux robots de maintenir des distances appropriées par rapport aux obstacles et aux personnes
FAQ
Comment calculer les distances entre des objets en utilisant Ultralytics YOLO26 ?
Pour calculer les distances entre des objets en utilisant Ultralytics YOLO26, tu dois identifier les centroïdes des bounding boxes des objets détectés. Ce processus implique l'initialisation de la classe DistanceCalculation depuis le module solutions d'Ultralytics et l'utilisation des sorties de suivi du modèle pour calculer les distances.
Quels sont les avantages d'utiliser le calcul de distance avec Ultralytics YOLO26 ?
L'utilisation du calcul de distance avec Ultralytics YOLO26 offre plusieurs avantages :
- Précision de localisation : Fournit un positionnement spatial précis pour les objets.
- Estimation de taille : Aide à estimer les tailles physiques, contribuant à une meilleure compréhension contextuelle.
- Compréhension de la scène : Améliore la compréhension des scènes 3D, facilitant une meilleure prise de décision dans des applications telles que la conduite autonome et la surveillance.
- Traitement en temps réel : Effectue les calculs à la volée, ce qui le rend adapté à l'analyse vidéo en direct.
- Capacités d'intégration : Fonctionne de manière transparente avec d'autres solutions YOLO26 comme le suivi d'objets et l'estimation de vitesse.
Puis-je effectuer un calcul de distance dans des flux vidéo en temps réel avec Ultralytics YOLO26 ?
Oui, tu peux effectuer un calcul de distance dans des flux vidéo en temps réel avec Ultralytics YOLO26. Le processus implique la capture d'images vidéo à l'aide d'OpenCV, l'exécution de la détection d'objets YOLO26 et l'utilisation de la classe DistanceCalculation pour calculer les distances entre les objets dans des images successives. Pour une mise en œuvre détaillée, voir l'exemple de flux vidéo.
Comment supprimer les points tracés pendant le calcul de distance en utilisant Ultralytics YOLO26 ?
Pour supprimer les points tracés pendant le calcul de distance avec Ultralytics YOLO26, tu peux utiliser un clic droit de la souris. Cette action effacera tous les points que tu as tracés. Pour plus de détails, reporte-toi à la section des remarques sous l'exemple de calcul de distance.
Quels sont les principaux arguments pour initialiser la classe DistanceCalculation dans Ultralytics YOLO26 ?
Les principaux arguments pour initialiser la classe DistanceCalculation dans Ultralytics YOLO26 incluent :
model: Chemin vers le fichier de modèle YOLO26.tracker: Algorithme de suivi à utiliser (la valeur par défaut est 'botsort.yaml').conf: Seuil de confiance pour les détections.show: Indicateur pour afficher la sortie.
Pour une liste exhaustive et les valeurs par défaut, voir les arguments de DistanceCalculation.
