import torch
import cv2
from PIL import Image
from time import sleep, time
import numpy as np
# Model
model = torch.hub.load('ultralytics/yolov5', 'custom', path_or_model='last.pt')
# Images
img = Image.open('test.png')
# convert image to numpy array
img = np.asarray(img)

# Inference
stime = time()
results = model(img, size=608)
print(time() - stime) #0.8 sec