From d4660ac7a3fddce2ab7025501a71ec0aa6dbc1e4 Mon Sep 17 00:00:00 2001
From: Ussama Naal <ussama.naal@ouster.io>
Date: Fri, 19 May 2023 11:14:48 -0700
Subject: [PATCH] Pip install ultralytics before requesting the model from
 torch hub

---
 Ouster_Yolo5_Demo.ipynb | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/Ouster_Yolo5_Demo.ipynb b/Ouster_Yolo5_Demo.ipynb
index 392bc00..238c562 100644
--- 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"