From d5f543a1dfa6b0c91d91664b6b35f7f4406e9726 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 16 Aug 2022 13:36:37 +0200 Subject: [PATCH] Created using Colaboratory --- tutorial.ipynb | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) 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": []