from PIL import Image | from PIL import Image | ||||
# Model | # Model | ||||
model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True).fuse().eval() # yolov5s.pt | |||||
model = model.autoshape() # for autoshaping of PIL/cv2/np inputs and NMS | |||||
model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True).autoshape() # for PIL/cv2/np inputs and NMS | |||||
# Images | # Images | ||||
img1 = Image.open('zidane.jpg') | img1 = Image.open('zidane.jpg') |