YOLO Object Detection mit Video

Question

Wie kann ich mit YOLOv5 Bilderkennung in einem Video machen?

import torch

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

results = model("test.mp4")

results.show()

Answer 1

Du kannst das Video mit OpenCV öffnen und das Model auf jeden Frame anwenden.

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()