Created using Colaboratory

tutorial.ipynb

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