Skip to content
Prev Previous commit
Next Next commit
update YOLOv9 notebook with better dataset version handling
  • Loading branch information
SkalskiP authored and ShreyasJoshi7 committed Mar 21, 2024
commit 5c67aaf12b058d71a3c5e405ac787080020ea1db
30 changes: 23 additions & 7 deletions notebooks/train-yolov9-object-detection-on-custom-dataset.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -445,9 +445,27 @@
{
"cell_type": "markdown",
"source": [
"**NOTE:** The dataset must be saved inside the `{HOME}/yolov9` directory, otherwise, the training will not succeed.\n",
"\n",
"In this tutorial, I will use the [football-players-detection](https://universe.roboflow.com/roboflow-jvuqo/football-players-detection-3zvbc) dataset. Feel free to replace it with your dataset in YOLO format or use another dataset available on [Roboflow Universe](https://universe.roboflow.com)."
"**NOTE:** The dataset must be saved inside the `{HOME}/yolov9` directory, otherwise, the training will not succeed."
],
"metadata": {
"id": "J5yx2GkI2P7Q"
}
},
{
"cell_type": "code",
"source": [
"%cd {HOME}/yolov9"
],
"metadata": {
"id": "MyLpftfU2Q1U"
},
"execution_count": null,
"outputs": []
},
{
"cell_type": "markdown",
"source": [
"**NOTE:** In this tutorial, I will use the [football-players-detection](https://universe.roboflow.com/roboflow-jvuqo/football-players-detection-3zvbc) dataset. Feel free to replace it with your dataset in YOLO format or use another dataset available on [Roboflow Universe](https://universe.roboflow.com). Additionally, if you plan to deploy your model to Roboflow after training, make sure you are the owner of the dataset and that no model is associated with the version of the dataset you are going to training on."
],
"metadata": {
"id": "eosmGt89vMO1"
Expand All @@ -456,16 +474,15 @@
{
"cell_type": "code",
"source": [
"%cd {HOME}/yolov9\n",
"\n",
"import roboflow\n",
"\n",
"roboflow.login()\n",
"\n",
"rf = roboflow.Roboflow()\n",
"\n",
"project = rf.workspace(\"roboflow-jvuqo\").project(\"football-players-detection-3zvbc\")\n",
"dataset = project.version(6).download(\"yolov9\")"
"version = project.version(6)\n",
"dataset = version.download(\"yolov9\")"
],
"metadata": {
"id": "4J3s_2_7p_gn"
Expand Down Expand Up @@ -845,7 +862,6 @@
{
"cell_type": "code",
"source": [
"version = project.version(6)\n",
"version.deploy(model_type=\"yolov9\", model_path=f\"{HOME}/yolov9/runs/train/exp\")"
],
"metadata": {
Expand Down