diff --git a/README.md b/README.md index 3de9271232c1..27ea18da1932 100755 --- a/README.md +++ b/README.md @@ -121,19 +121,18 @@ Done. (0.103s) ### PyTorch Hub -To run **batched inference** with YOLOv5 and [PyTorch Hub](https://github.com/ultralytics/yolov5/issues/36): +Inference with YOLOv5 and [PyTorch Hub](https://github.com/ultralytics/yolov5/issues/36): ```python import torch # Model model = torch.hub.load('ultralytics/yolov5', 'yolov5s') -# Images -dir = 'https://github.com/ultralytics/yolov5/raw/master/data/images/' -imgs = [dir + f for f in ('zidane.jpg', 'bus.jpg')] # batch of images +# Image +img = 'https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg' # Inference -results = model(imgs) +results = model(img) results.print() # or .show(), .save() ```