From 8c0e7092a28a38beedb0bd21dfc1b46713adf568 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 23 Nov 2020 19:28:35 +0100 Subject: [PATCH] Update README.md --- README.md | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/README.md b/README.md index 43e480d..72b3c37 100755 --- a/README.md +++ b/README.md @@ -108,8 +108,7 @@ import torch from PIL import Image # 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 img1 = Image.open('zidane.jpg')