Yolo Object Detection mit Webcam

Question

Ich habe dieses Script gefunden um Objekte in einem Video mit Yolo und OpenCV zu erkennen.

import torch
import cv2

model = torch.hub.load('ultralytics/yolov5', 'yolov5n')

cap = cv2.VideoCapture("test.mp4")

while cap.isOpened():
    
    ret, frame = cap.read()

    results = model(frame)

    cv2.imshow('Object Detector', results.render()[0])

    key = cv2.waitKey(1)

cap.release()
cv2.destroyAllWindows()

Kann ich das auch mit meiner Webcam machen?

Answer 1

Du musst das Eingabegerät anpassen, z.B

cap = cv2.VideoCapture(0)

für das erste Device u.s.w.