• Agent DVR

import torch
import cv2
 
model = torch.hub.load('ultralytics/yolov5', 'yolov5n')
 
cap = cv2.VideoCapture("test.mp4")
# cap = cv2.VideoCapture(0) # First device in list
 
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()