Python

Rastreo de objetos YOLO

Vista de consulta del código. Los datos sensibles en URLs se ocultan en esta vista.

# Rastreo de objetos YOLO
import cv2
from ultralytics import YOLO 
model=YOLO("yolo11n")
cap=cv2.VideoCapture("imgs/carros.jpeg")
#cap=cv2.VideoCapture(1)

while True:
    status,frame=cap.read()
    if not status:
        break  
    results=model.track(frame,conf=.2,classes=[2],verbose=False,persist=True,tracker="botsort.yaml")  
    frame=results[0].plot(line_width=2)
    cv2.imshow("frame",frame)  
    cv2.waitKey(0)
    if cv2.waitKey(1) & 0xFF==ord(" "):
        cv2.waitKey()  
    if cv2.waitKey(1) & 0xFF==ord("q"):
        break
cap.release()