update ultralytics demo

Author: fcakyon

README.md

@@ -21,7 +21,7 @@
   <br>
     <a href="https://ieeexplore.ieee.org/document/9897990"><img src="https://img.shields.io/badge/DOI-10.1109%2FICIP46576.2022.9897990-orange.svg" alt="ci"></a>
   <br>
-    <a href="https://colab.research.google.com/github/obss/sahi/blob/main/demo/inference_for_yolov5.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"></a>
+    <a href="https://colab.research.google.com/github/obss/sahi/blob/main/demo/inference_for_ultralytics.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"></a>
     <a href="https://huggingface.co/spaces/fcakyon/sahi-yolox"><img src="https://raw.githubusercontent.com/obss/sahi/main/resources/hf_spaces_badge.svg" alt="HuggingFace Spaces"></a>
 
 </div>

demo/inference_for_ultralytics.ipynb

@@ -4,7 +4,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/obss/sahi/blob/main/demo/inference_for_yolov5.ipynb)"
+    "[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/obss/sahi/blob/main/demo/inference_for_ultralytics.ipynb)"
    ]
   },
   {
@@ -81,7 +81,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "yolo11n_model_path = \"models/yolov11n.pt\"\n",
+    "yolo11n_model_path = \"models/yolov11n.pt\" # or change to any local Ultralytics model path\n",
     "download_yolo11n_model(yolo11n_model_path)\n",
     "# yolov8n_model_path = \"models/yolov8n.pt\"\n",
     "# download_yolov8n_model(yolov8n_model_path)\n",
@@ -113,9 +113,9 @@
    "outputs": [],
    "source": [
     "detection_model = AutoDetectionModel.from_pretrained(\n",
-    "    model_type='yolo11', # or 'yolov8'\n",
+    "    model_type='ultralytics',\n",
     "    model_path=yolo11n_model_path,\n",
-    "    confidence_threshold=0.3,\n",
+    "    confidence_threshold=0.35,\n",
     "    device=\"cpu\", # or 'cuda:0'\n",
     ")"
    ]