Merge pull request #2 from ouster-lidar/pip-install-ultralytics

Pip install ultralytics before requesting the model from torch hub
ouster-lidar · May 19, 2023 · 5740e0c · 5740e0c
2 parents c0355d3 + d4660ac
commit 5740e0c
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/Ouster_Yolo5_Demo.ipynb b/Ouster_Yolo5_Demo.ipynb
@@ -1,6 +1,7 @@
 {
   "cells": [
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "colab_type": "text",
@@ -11,6 +12,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "hrsaDfdVHzxt"
@@ -31,6 +33,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "yNveqeA1KXGy"
@@ -71,6 +74,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "gSoLdnyNvkKH"
@@ -111,6 +115,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "X7yAi9hd-T4B"
@@ -177,6 +182,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "zP6USLgz2f0r"
@@ -203,6 +209,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "MlSPiCVaFGRR"
@@ -259,6 +266,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "jtmS7_TXFsT3"
@@ -307,10 +315,12 @@
       ],
       "source": [
         "# Load YOLOv5 model\n",
+        "%pip install ultralytics\n",
         "model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)\n"
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "JzytMgGJc8yi"
@@ -383,6 +393,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "cdjANOTc-QiI"
@@ -424,6 +435,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "sBQZO1YkubR2"
@@ -486,6 +498,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "DHkrGN4OPv-S"
@@ -613,6 +626,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "g8dHcni6CJYt"