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
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.

Calcul de distance avec Ultralytics YOLO
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 avec les arguments de DistanceCalculation :

ArgumentTypeDéfautDescription
modelstrNoneChemin d'accès vers un fichier de modèle Ultralytics YOLO.

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

ArgumentTypeDéfautDescription
trackerstr'botsort.yaml'Spécifie l'algorithme de suivi à utiliser, par exemple bytetrack.yaml ou botsort.yaml.
conffloat0.1Dé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.
ioufloat0.7Définit le seuil Intersection over Union (IoU) pour filtrer les détections qui se chevauchent.
classeslistNoneFiltre les résultats par index de classe. Par exemple, classes=[0, 2, 3] suit uniquement les classes spécifiées.
verboseboolTrueContrôle l'affichage des résultats de suivi, fournissant une sortie visuelle des objets suivis.
devicestrNoneSpé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 :

ArgumentTypeDéfautDescription
showboolFalseSi 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_widthint or NoneNoneSpé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_confboolTrueAffiche 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_labelsboolTrueAffiche 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 :

  1. Extrait les centroïdes (points centraux) des bounding boxes sélectionnées
  2. Calcule la distance euclidienne entre ces centroïdes en pixels
  3. 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.

Commentaires