Merge pull request #354 from sitamgithub-MSIT/yolo11-fix

Yolo11 finetuning notebooks path fix for kaggle
roboflow · Dec 16, 2024 · 94b1cb2 · 94b1cb2
2 parents 16d9fd5 + 48912fb
commit 94b1cb2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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\")"