update YOLOv9 notebook

roboflow · Mar 1, 2024 · f4ffff8 · f4ffff8
1 parent ab2325d
commit f4ffff8
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/notebooks/train-yolov9-object-detection-on-custom-dataset.ipynb b/notebooks/train-yolov9-object-detection-on-custom-dataset.ipynb
@@ -481,7 +481,7 @@
         "rf = roboflow.Roboflow()\n",
         "\n",
         "project = rf.workspace(\"roboflow-jvuqo\").project(\"football-players-detection-3zvbc\")\n",
-        "version = project.version(6)\n",
+        "version = project.version(8)\n",
         "dataset = version.download(\"yolov9\")"
       ],
       "metadata": {
@@ -732,7 +732,7 @@
         "!python detect.py \\\n",
         "--img 1280 --conf 0.1 --device 0 \\\n",
         "--weights {HOME}/yolov9/runs/train/exp/weights/best.pt \\\n",
-        "--source {dataset.location}/valid/images"
+        "--source {dataset.location}/test/images"
       ],
       "metadata": {
         "id": "8vnrn9cwIsUs"
@@ -899,7 +899,7 @@
       "source": [
         "ROBOFLOW_API_KEY = getpass.getpass()\n",
         "\n",
-        "model = get_model(model_id=\"football-players-detection-3zvbc/6\", api_key=ROBOFLOW_API_KEY)"
+        "model = get_model(model_id=\"football-players-detection-3zvbc/8\", api_key=ROBOFLOW_API_KEY)"
       ],
       "metadata": {
         "id": "bAB-5ZMM87w3",
@@ -922,7 +922,7 @@
     {
       "cell_type": "markdown",
       "source": [
-        "**NOTE:** Let's pick random image from our validation subset and detect objects using newly fine-tuned model."
+        "**NOTE:** Let's pick random image from our test subset and detect objects using newly fine-tuned model."
       ],
       "metadata": {
         "id": "5pGSLZ8Fz5qO"
@@ -932,7 +932,7 @@
       "cell_type": "code",
       "source": [
         "image_paths = sv.list_files_with_extensions(\n",
-        "    directory=f\"{dataset.location}/valid/images\",\n",
+        "    directory=f\"{dataset.location}/test/images\",\n",
         "    extensions=['png', 'jpg', 'jpeg']\n",
         ")\n",
         "image_path = random.choice(image_paths)\n",