import cv2
import torch
import numpy as np

# model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained = True, force_reload = True)
model = torch.hub.load('ultralytics/yolov5', 'custom', path = 'yolov5s.pt', force_reload = True)

# cap = cv2.VideoCapture(0)
cap = cv2.VideoCapture(0)

while cap.isOpened():
    ret, frame = cap.read()
    results = model(frame)
    cv2.imshow('frame',  np.squeeze(results.render()))
      
    if cv2.waitKey(1) == 27:
        break
  
cap.release()
cv2.destroyAllWindows()