From 48912fb601fec7237cec2e1aee5211b3122c04d7 Mon Sep 17 00:00:00 2001
From: Sitam Meur <103279526+sitamgithub-MSIT@users.noreply.github.com>
Date: Sun, 15 Dec 2024 12:55:14 +0000
Subject: [PATCH] yolo11 obj detection path fix

---
 notebooks/train-yolo11-object-detection-on-custom-dataset.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/notebooks/train-yolo11-object-detection-on-custom-dataset.ipynb b/notebooks/train-yolo11-object-detection-on-custom-dataset.ipynb
index de3f6c8..8f13029 100644
--- a/notebooks/train-yolo11-object-detection-on-custom-dataset.ipynb
+++ b/notebooks/train-yolo11-object-detection-on-custom-dataset.ipynb
@@ -880,7 +880,7 @@
         "import os\n",
         "from IPython.display import Image as IPyImage, display\n",
         "\n",
-        "latest_folder = max(glob.glob('/content/runs/detect/predict*/'), key=os.path.getmtime)\n",
+        "latest_folder = max(glob.glob(f'{HOME}/runs/detect/predict*/'), key=os.path.getmtime)\n",
         "for img in glob.glob(f'{latest_folder}/*.jpg')[:3]:\n",
         "    display(IPyImage(filename=img, width=600))\n",
         "    print(\"\\n\")"