diff --git a/tutorial.ipynb b/tutorial.ipynb index da893c3..c39faab 100644 --- a/tutorial.ipynb +++ b/tutorial.ipynb @@ -1023,19 +1023,19 @@ "id": "GMusP4OAxFu6" }, "source": [ - "# PyTorch Hub\n", "import torch\n", "\n", - "# Model\n", - "model = torch.hub.load('ultralytics/yolov5', 'yolov5s')\n", + "# PyTorch Hub Model\n", + "model = torch.hub.load('ultralytics/yolov5', 'yolov5s') # or yolov5n - yolov5x6, custom\n", "\n", "# Images\n", - "dir = 'https://ultralytics.com/images/'\n", - "imgs = [dir + f for f in ('zidane.jpg', 'bus.jpg')] # batch of images\n", + "img = 'https://ultralytics.com/images/zidane.jpg' # or file, Path, PIL, OpenCV, numpy, list\n", "\n", "# Inference\n", - "results = model(imgs)\n", - "results.print() # or .show(), .save()" + "results = model(img)\n", + "\n", + "# Results\n", + "results.print() # or .show(), .save(), .crop(), .pandas(), etc." ], "execution_count": null, "outputs": []