Calcul de distance avec Ultralytics YOLO26#

Qu'est-ce que le calcul de distance ?#

Le calcul de distance est le processus qui consiste à mesurer l'espace entre deux objets détectés dans une image ou une trame vidéo. Dans le cas de Ultralytics YOLO26, le centroïde de la boîte englobante est utilisé pour calculer la distance entre les boîtes englobantes mises en évidence par l'utilisateur.

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 la taille : permet d'estimer la taille des objets pour une meilleure compréhension du contexte.
• Compréhension de la scène : améliore la compréhension des scènes 3D afin de favoriser une meilleure prise de décision dans des applications telles que les véhicules autonomes et les systèmes de surveillance.
• Évitement des collisions : 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 dans l'environnement surveillé.

Calculer les distances avec YOLO26#

La solution DistanceCalculation suit les objets d'une trame à l'autre et mesure, en pixels, la distance euclidienne entre les centroïdes de deux boîtes englobantes quelconques que tu sélectionnes avec la souris. Exécute l'exemple ci-dessous, puis clique avec le bouton gauche sur deux objets pour tracer la ligne qui les relie et afficher la distance ; clique avec le bouton droit pour effacer ta sélection.

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 windows

Arguments de DistanceCalculation()#

Voici un tableau présentant les arguments de DistanceCalculation :

Tu peux également utiliser différents arguments de track dans la solution DistanceCalculation.

De plus, les arguments de visualisation suivants sont disponibles :

Détails de l'implémentation#

La classe DistanceCalculation fonctionne en suivant les objets d'une trame vidéo à l'autre et en calculant la distance euclidienne entre les centroïdes des boîtes englobantes sélectionnées. Lorsque tu cliques sur deux objets, la solution :

1. extrait les centroïdes (points centraux) des boîtes englobantes sélectionnées ;
2. calcule la distance euclidienne entre ces centroïdes en pixels ;
3. affiche la distance sur la trame avec une ligne reliant les objets ;

L'implémentation utilise la méthode mouse_event_for_distance pour gérer les interactions avec la souris, ce qui permet aux utilisateurs de sélectionner des objets et d'effacer les sélections selon leurs besoins. La méthode process gère le traitement trame par trame, le suivi des objets et le calcul des distances.

Applications#

Le calcul de distance avec YOLO26 possède de nombreuses applications pratiques :

• Analyse commerciale : mesure la proximité des clients par rapport aux 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 entre les véhicules et détecte le non-respect des distances de sécurité
• Analyse sportive : calcule les distances entre les joueurs, le ballon et les positions clés du terrain
• Santé : garantit une distance appropriée dans les salles d'attente et surveille les déplacements des patients
• Robotique : permet aux robots de maintenir des distances appropriées par rapport aux obstacles et aux personnes

FAQ#