updated

.gitattributes

@@ -61,3 +61,6 @@ runs/detect/yolov11m_1920p/weights/epoch10.pt filter=lfs diff=lfs merge=lfs -tex
 runs/detect/yolov11m_1920p/weights/epoch5.pt filter=lfs diff=lfs merge=lfs -text
 runs/detect/yolov11m_1920p/weights/last.pt filter=lfs diff=lfs merge=lfs -text
 runs/detect/yolov11m_1920p/weights/best.pt filter=lfs diff=lfs merge=lfs -text
+notebook_images/cots_dataset.png filter=lfs diff=lfs merge=lfs -text
+notebook_images/yolo_dataset2.png filter=lfs diff=lfs merge=lfs -text
+notebook_images/yolo_dataset.png filter=lfs diff=lfs merge=lfs -text

cots_great_barrier_reef_ultralytics.ipynb

@@ -83,7 +83,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 1,
    "id": "0171cc0b-4425-4cb7-b084-0f406e58a287",
    "metadata": {},
    "outputs": [],
@@ -95,7 +95,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 2,
    "id": "efd9a0ea-a343-4955-87ac-0b7635de709c",
    "metadata": {},
    "outputs": [],
@@ -112,7 +112,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 3,
    "id": "6165ae5f-3ff9-438b-8057-562d6c44ab08",
    "metadata": {},
    "outputs": [
@@ -120,7 +120,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "/home/light/personal_projects/vision_projects/notebooks\n"
+      "/home/light/personal_projects/cots_detection_project\n"
      ]
     }
    ],
@@ -141,7 +141,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 4,
    "id": "84ff5bb0-d24c-48b0-a805-6a62a0dbab35",
    "metadata": {},
    "outputs": [],
@@ -167,7 +167,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 5,
    "id": "3c5ac0f5-3895-43cc-83d3-77aabc234e99",
    "metadata": {},
    "outputs": [],
@@ -179,7 +179,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": 6,
    "id": "b641e9b9-230e-4686-8283-aa305b5b7ebc",
    "metadata": {},
    "outputs": [
@@ -192,7 +192,7 @@
        "dtype: object"
       ]
      },
-     "execution_count": 8,
+     "execution_count": 6,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -203,7 +203,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": 7,
    "id": "13022f50-cd2a-49d3-a13e-13f982196db1",
    "metadata": {},
    "outputs": [
@@ -214,7 +214,7 @@
        " {'id': 2, 'name': 'SCAR', 'supercategory': ''}]"
       ]
      },
-     "execution_count": 9,
+     "execution_count": 7,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -225,7 +225,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": 8,
    "id": "480e5363-7679-4a6c-b22b-6be05b747cdc",
    "metadata": {},
    "outputs": [
@@ -243,7 +243,7 @@
        " 'source': 'REEFSCAN_DEEP_02_20241026_020155_Judith'}"
       ]
      },
-     "execution_count": 10,
+     "execution_count": 8,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -254,7 +254,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": 9,
    "id": "b32bdf4d-9499-4aff-b18c-b93100d62a1f",
    "metadata": {},
    "outputs": [
@@ -298,7 +298,7 @@
        " 'attributes': {'comment': '', 'type': '', 'occluded': False}}"
       ]
      },
-     "execution_count": 11,
+     "execution_count": 9,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -317,7 +317,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 12,
+   "execution_count": 10,
    "id": "e6049d45-0b03-4a1e-bae1-bb64243fe300",
    "metadata": {},
    "outputs": [],
@@ -334,7 +334,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 13,
+   "execution_count": 11,
    "id": "25dd1dfd-d8f6-45f4-9760-f4f638698cb1",
    "metadata": {},
    "outputs": [
@@ -368,7 +368,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 14,
+   "execution_count": 12,
    "id": "7a999041-427e-428a-9d02-ffc90a97543a",
    "metadata": {},
    "outputs": [
@@ -400,7 +400,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 15,
+   "execution_count": 13,
    "id": "f1337db2-6ae5-4638-b98f-1e18aa7651a7",
    "metadata": {},
    "outputs": [
@@ -445,7 +445,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 16,
+   "execution_count": 14,
    "id": "0dc8660e-a70b-471e-bd90-ee379e9e7314",
    "metadata": {},
    "outputs": [],
@@ -455,7 +455,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 17,
+   "execution_count": 15,
    "id": "2d0bc42a-e714-46de-b9b5-9df97ec4d887",
    "metadata": {},
    "outputs": [
@@ -483,7 +483,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 18,
+   "execution_count": 16,
    "id": "4e0b3ae1-ba85-4673-a7c7-c4ffb11b0d20",
    "metadata": {},
    "outputs": [],
@@ -500,7 +500,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 19,
+   "execution_count": 17,
    "id": "d28f08ee-ae68-45e6-bf7a-b638f81cf3b8",
    "metadata": {},
    "outputs": [],
@@ -530,7 +530,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 20,
+   "execution_count": 18,
    "id": "a01c9f06-29d6-4f8b-9baa-95cc9e6dbede",
    "metadata": {},
    "outputs": [],
@@ -566,7 +566,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 21,
+   "execution_count": 19,
    "id": "e141220f-cd5f-4646-a417-1d31f97faf86",
    "metadata": {},
    "outputs": [],
@@ -627,10 +627,18 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 20,
    "id": "8c4c2f72-fd88-4225-96cb-64f078df7e0e",
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Using device cuda\n"
+     ]
+    }
+   ],
    "source": [
     "import torch\n",
     "import torchvision\n",
@@ -651,7 +659,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 22,
+   "execution_count": 21,
    "id": "80a80e04-3904-4396-becd-aea967e5135c",
    "metadata": {},
    "outputs": [
@@ -660,55 +668,55 @@
      "output_type": "stream",
      "text": [
       "Requirement already satisfied: ultralytics in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (8.3.161)\n",
-      "Requirement already satisfied: scipy>=1.4.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (1.13.1)\n",
-      "Requirement already satisfied: opencv-python>=4.6.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (4.11.0.86)\n",
-      "Requirement already satisfied: ultralytics-thop>=2.0.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (2.0.14)\n",
+      "Requirement already satisfied: pyyaml>=5.3.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (6.0.2)\n",
+      "Requirement already satisfied: requests>=2.23.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (2.32.4)\n",
+      "Requirement already satisfied: pillow>=7.1.2 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (10.4.0)\n",
       "Requirement already satisfied: torch>=1.8.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (2.7.1)\n",
-      "Requirement already satisfied: psutil in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (7.0.0)\n",
       "Requirement already satisfied: torchvision>=0.9.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (0.22.1)\n",
       "Requirement already satisfied: py-cpuinfo in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (9.0.0)\n",
-      "Requirement already satisfied: pillow>=7.1.2 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (10.4.0)\n",
-      "Requirement already satisfied: requests>=2.23.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (2.32.4)\n",
-      "Requirement already satisfied: matplotlib>=3.3.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (3.9.4)\n",
-      "Requirement already satisfied: numpy>=1.23.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (2.0.2)\n",
+      "Requirement already satisfied: ultralytics-thop>=2.0.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (2.0.14)\n",
       "Requirement already satisfied: pandas>=1.1.4 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (2.3.0)\n",
-      "Requirement already satisfied: pyyaml>=5.3.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (6.0.2)\n",
+      "Requirement already satisfied: numpy>=1.23.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (2.0.2)\n",
+      "Requirement already satisfied: scipy>=1.4.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (1.13.1)\n",
+      "Requirement already satisfied: opencv-python>=4.6.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (4.11.0.86)\n",
       "Requirement already satisfied: tqdm>=4.64.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (4.67.1)\n",
-      "Requirement already satisfied: cycler>=0.10 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.3.0->ultralytics) (0.12.1)\n",
-      "Requirement already satisfied: pyparsing>=2.3.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.3.0->ultralytics) (3.2.3)\n",
-      "Requirement already satisfied: kiwisolver>=1.3.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.3.0->ultralytics) (1.4.7)\n",
-      "Requirement already satisfied: packaging>=20.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.3.0->ultralytics) (25.0)\n",
+      "Requirement already satisfied: matplotlib>=3.3.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (3.9.4)\n",
+      "Requirement already satisfied: psutil in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from ultralytics) (7.0.0)\n",
       "Requirement already satisfied: contourpy>=1.0.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.3.0->ultralytics) (1.3.0)\n",
-      "Requirement already satisfied: fonttools>=4.22.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.3.0->ultralytics) (4.58.4)\n",
       "Requirement already satisfied: importlib-resources>=3.2.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.3.0->ultralytics) (6.5.2)\n",
+      "Requirement already satisfied: packaging>=20.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.3.0->ultralytics) (25.0)\n",
+      "Requirement already satisfied: kiwisolver>=1.3.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.3.0->ultralytics) (1.4.7)\n",
+      "Requirement already satisfied: cycler>=0.10 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.3.0->ultralytics) (0.12.1)\n",
       "Requirement already satisfied: python-dateutil>=2.7 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.3.0->ultralytics) (2.9.0.post0)\n",
-      "Requirement already satisfied: pytz>=2020.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from pandas>=1.1.4->ultralytics) (2025.2)\n",
+      "Requirement already satisfied: pyparsing>=2.3.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.3.0->ultralytics) (3.2.3)\n",
+      "Requirement already satisfied: fonttools>=4.22.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.3.0->ultralytics) (4.58.4)\n",
       "Requirement already satisfied: tzdata>=2022.7 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from pandas>=1.1.4->ultralytics) (2025.2)\n",
+      "Requirement already satisfied: pytz>=2020.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from pandas>=1.1.4->ultralytics) (2025.2)\n",
+      "Requirement already satisfied: charset_normalizer<4,>=2 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from requests>=2.23.0->ultralytics) (3.4.2)\n",
       "Requirement already satisfied: certifi>=2017.4.17 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from requests>=2.23.0->ultralytics) (2025.6.15)\n",
       "Requirement already satisfied: urllib3<3,>=1.21.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from requests>=2.23.0->ultralytics) (2.5.0)\n",
-      "Requirement already satisfied: charset_normalizer<4,>=2 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from requests>=2.23.0->ultralytics) (3.4.2)\n",
       "Requirement already satisfied: idna<4,>=2.5 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from requests>=2.23.0->ultralytics) (3.10)\n",
-      "Requirement already satisfied: nvidia-cudnn-cu12==9.5.1.17 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (9.5.1.17)\n",
-      "Requirement already satisfied: fsspec in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (2025.5.1)\n",
-      "Requirement already satisfied: nvidia-cufft-cu12==11.3.0.4 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (11.3.0.4)\n",
-      "Requirement already satisfied: nvidia-cuda-nvrtc-cu12==12.6.77 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (12.6.77)\n",
+      "Requirement already satisfied: triton==3.3.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (3.3.1)\n",
+      "Requirement already satisfied: nvidia-cuda-cupti-cu12==12.6.80 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (12.6.80)\n",
+      "Requirement already satisfied: nvidia-cublas-cu12==12.6.4.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (12.6.4.1)\n",
+      "Requirement already satisfied: jinja2 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (3.1.6)\n",
+      "Requirement already satisfied: filelock in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (3.18.0)\n",
+      "Requirement already satisfied: nvidia-cusparse-cu12==12.5.4.2 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (12.5.4.2)\n",
       "Requirement already satisfied: nvidia-nvtx-cu12==12.6.77 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (12.6.77)\n",
-      "Requirement already satisfied: networkx in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (3.2.1)\n",
-      "Requirement already satisfied: nvidia-cusolver-cu12==11.7.1.2 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (11.7.1.2)\n",
+      "Requirement already satisfied: nvidia-nccl-cu12==2.26.2 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (2.26.2)\n",
       "Requirement already satisfied: nvidia-cufile-cu12==1.11.1.6 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (1.11.1.6)\n",
+      "Requirement already satisfied: nvidia-curand-cu12==10.3.7.77 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (10.3.7.77)\n",
+      "Requirement already satisfied: nvidia-cuda-runtime-cu12==12.6.77 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (12.6.77)\n",
       "Requirement already satisfied: nvidia-cusparselt-cu12==0.6.3 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (0.6.3)\n",
-      "Requirement already satisfied: jinja2 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (3.1.6)\n",
       "Requirement already satisfied: typing-extensions>=4.10.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (4.14.1)\n",
-      "Requirement already satisfied: filelock in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (3.18.0)\n",
-      "Requirement already satisfied: nvidia-cublas-cu12==12.6.4.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (12.6.4.1)\n",
+      "Requirement already satisfied: networkx in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (3.2.1)\n",
       "Requirement already satisfied: nvidia-nvjitlink-cu12==12.6.85 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (12.6.85)\n",
-      "Requirement already satisfied: triton==3.3.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (3.3.1)\n",
-      "Requirement already satisfied: nvidia-cuda-cupti-cu12==12.6.80 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (12.6.80)\n",
-      "Requirement already satisfied: nvidia-cuda-runtime-cu12==12.6.77 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (12.6.77)\n",
-      "Requirement already satisfied: nvidia-cusparse-cu12==12.5.4.2 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (12.5.4.2)\n",
-      "Requirement already satisfied: nvidia-nccl-cu12==2.26.2 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (2.26.2)\n",
+      "Requirement already satisfied: nvidia-cusolver-cu12==11.7.1.2 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (11.7.1.2)\n",
+      "Requirement already satisfied: nvidia-cudnn-cu12==9.5.1.17 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (9.5.1.17)\n",
+      "Requirement already satisfied: nvidia-cufft-cu12==11.3.0.4 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (11.3.0.4)\n",
+      "Requirement already satisfied: fsspec in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (2025.3.0)\n",
+      "Requirement already satisfied: nvidia-cuda-nvrtc-cu12==12.6.77 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (12.6.77)\n",
       "Requirement already satisfied: sympy>=1.13.3 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (1.14.0)\n",
-      "Requirement already satisfied: nvidia-curand-cu12==10.3.7.77 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from torch>=1.8.0->ultralytics) (10.3.7.77)\n",
       "Requirement already satisfied: setuptools>=40.8.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from triton==3.3.1->torch>=1.8.0->ultralytics) (58.1.0)\n",
       "Requirement already satisfied: zipp>=3.1.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from importlib-resources>=3.2.0->matplotlib>=3.3.0->ultralytics) (3.23.0)\n",
       "Requirement already satisfied: six>=1.5 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from python-dateutil>=2.7->matplotlib>=3.3.0->ultralytics) (1.17.0)\n",
@@ -726,7 +734,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 23,
+   "execution_count": 22,
    "id": "6596c2c4-3413-4e98-b4e1-929993927db7",
    "metadata": {},
    "outputs": [
@@ -735,18 +743,18 @@
      "output_type": "stream",
      "text": [
       "Requirement already satisfied: albumentations in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (2.0.8)\n",
-      "Requirement already satisfied: eval-type-backport in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from albumentations) (0.2.2)\n",
-      "Requirement already satisfied: albucore==0.0.24 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from albumentations) (0.0.24)\n",
-      "Requirement already satisfied: pydantic>=2.9.2 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from albumentations) (2.10.6)\n",
-      "Requirement already satisfied: opencv-python-headless>=4.9.0.80 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from albumentations) (4.11.0.86)\n",
-      "Requirement already satisfied: typing-extensions>=4.9.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from albumentations) (4.14.1)\n",
       "Requirement already satisfied: scipy>=1.10.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from albumentations) (1.13.1)\n",
       "Requirement already satisfied: PyYAML in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from albumentations) (6.0.2)\n",
+      "Requirement already satisfied: typing-extensions>=4.9.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from albumentations) (4.14.1)\n",
+      "Requirement already satisfied: pydantic>=2.9.2 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from albumentations) (2.10.6)\n",
       "Requirement already satisfied: numpy>=1.24.4 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from albumentations) (2.0.2)\n",
-      "Requirement already satisfied: simsimd>=5.9.2 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from albucore==0.0.24->albumentations) (6.4.9)\n",
+      "Requirement already satisfied: albucore==0.0.24 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from albumentations) (0.0.24)\n",
+      "Requirement already satisfied: opencv-python-headless>=4.9.0.80 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from albumentations) (4.11.0.86)\n",
+      "Requirement already satisfied: eval-type-backport in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from albumentations) (0.2.2)\n",
       "Requirement already satisfied: stringzilla>=3.10.4 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from albucore==0.0.24->albumentations) (3.12.5)\n",
-      "Requirement already satisfied: pydantic-core==2.27.2 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from pydantic>=2.9.2->albumentations) (2.27.2)\n",
+      "Requirement already satisfied: simsimd>=5.9.2 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from albucore==0.0.24->albumentations) (6.4.9)\n",
       "Requirement already satisfied: annotated-types>=0.6.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from pydantic>=2.9.2->albumentations) (0.7.0)\n",
+      "Requirement already satisfied: pydantic-core==2.27.2 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from pydantic>=2.9.2->albumentations) (2.27.2)\n",
       "\n",
       "\u001b[1m[\u001b[0m\u001b[34;49mnotice\u001b[0m\u001b[1;39;49m]\u001b[0m\u001b[39;49m A new release of pip is available: \u001b[0m\u001b[31;49m23.0.1\u001b[0m\u001b[39;49m -> \u001b[0m\u001b[32;49m25.2\u001b[0m\n",
       "\u001b[1m[\u001b[0m\u001b[34;49mnotice\u001b[0m\u001b[1;39;49m]\u001b[0m\u001b[39;49m To update, run: \u001b[0m\u001b[32;49mpip install --upgrade pip\u001b[0m\n"
@@ -759,7 +767,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 24,
+   "execution_count": 23,
    "id": "f665634b-3409-4f95-8587-3141a9a1330d",
    "metadata": {},
    "outputs": [
@@ -768,16 +776,16 @@
      "output_type": "stream",
      "text": [
       "Requirement already satisfied: tensorboard in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (2.19.0)\n",
-      "Requirement already satisfied: packaging in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from tensorboard) (25.0)\n",
-      "Requirement already satisfied: absl-py>=0.4 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from tensorboard) (2.3.0)\n",
-      "Requirement already satisfied: werkzeug>=1.0.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from tensorboard) (3.1.3)\n",
       "Requirement already satisfied: setuptools>=41.0.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from tensorboard) (58.1.0)\n",
+      "Requirement already satisfied: protobuf!=4.24.0,>=3.19.6 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from tensorboard) (6.31.1)\n",
       "Requirement already satisfied: tensorboard-data-server<0.8.0,>=0.7.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from tensorboard) (0.7.2)\n",
-      "Requirement already satisfied: markdown>=2.6.8 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from tensorboard) (3.8.2)\n",
+      "Requirement already satisfied: absl-py>=0.4 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from tensorboard) (2.3.0)\n",
+      "Requirement already satisfied: numpy>=1.12.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from tensorboard) (2.0.2)\n",
       "Requirement already satisfied: grpcio>=1.48.2 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from tensorboard) (1.73.1)\n",
+      "Requirement already satisfied: packaging in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from tensorboard) (25.0)\n",
       "Requirement already satisfied: six>1.9 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from tensorboard) (1.17.0)\n",
-      "Requirement already satisfied: numpy>=1.12.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from tensorboard) (2.0.2)\n",
-      "Requirement already satisfied: protobuf!=4.24.0,>=3.19.6 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from tensorboard) (6.31.1)\n",
+      "Requirement already satisfied: markdown>=2.6.8 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from tensorboard) (3.8.2)\n",
+      "Requirement already satisfied: werkzeug>=1.0.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from tensorboard) (3.1.3)\n",
       "Requirement already satisfied: importlib-metadata>=4.4 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from markdown>=2.6.8->tensorboard) (8.7.0)\n",
       "Requirement already satisfied: MarkupSafe>=2.1.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from werkzeug>=1.0.1->tensorboard) (2.1.5)\n",
       "Requirement already satisfied: zipp>=3.20 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from importlib-metadata>=4.4->markdown>=2.6.8->tensorboard) (3.23.0)\n",
@@ -793,36 +801,19 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 47,
+   "execution_count": 24,
    "id": "ab88ba4b-2136-40f3-a711-d176298a275d",
    "metadata": {},
    "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "The tensorboard extension is already loaded. To reload it, use:\n",
-      "  %reload_ext tensorboard\n"
-     ]
-    },
-    {
-     "data": {
-      "text/plain": [
-       "Reusing TensorBoard on port 6006 (pid 103198), started 16:08:58 ago. (Use '!kill 103198' to kill it.)"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    },
     {
      "data": {
       "text/html": [
        "\n",
-       "      <iframe id=\"tensorboard-frame-50f244556f25e2a2\" width=\"100%\" height=\"800\" frameborder=\"0\">\n",
+       "      <iframe id=\"tensorboard-frame-f87b3da6e9f0de99\" width=\"100%\" height=\"800\" frameborder=\"0\">\n",
        "      </iframe>\n",
        "      <script>\n",
        "        (function() {\n",
-       "          const frame = document.getElementById(\"tensorboard-frame-50f244556f25e2a2\");\n",
+       "          const frame = document.getElementById(\"tensorboard-frame-f87b3da6e9f0de99\");\n",
        "          const url = new URL(\"/\", window.location);\n",
        "          const port = 6006;\n",
        "          if (port) {\n",
@@ -848,7 +839,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 26,
+   "execution_count": 25,
    "id": "c19d9310-340a-490e-8668-f2f9fdf1c094",
    "metadata": {},
    "outputs": [
@@ -907,7 +898,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 27,
+   "execution_count": 26,
    "id": "abb9c333-ce15-45b8-ac04-e3e57a882378",
    "metadata": {},
    "outputs": [],
@@ -1171,7 +1162,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 27,
    "id": "0e9bbc6d-4172-4a66-bc77-d9332a92dfb9",
    "metadata": {},
    "outputs": [
@@ -1181,7 +1172,7 @@
      "text": [
       "New https://pypi.org/project/ultralytics/8.3.176 available 😃 Update with 'pip install -U ultralytics'\n",
       "Ultralytics 8.3.161 🚀 Python-3.9.22 torch-2.7.1+cu126 CUDA:0 (NVIDIA GeForce RTX 4090 Laptop GPU, 15955MiB)\n",
-      "\u001b[34m\u001b[1mengine/trainer: \u001b[0magnostic_nms=False, amp=True, augment=False, auto_augment=randaugment, batch=3, bgr=0.0, box=7.5, cache=False, cfg=None, classes=None, close_mosaic=10, cls=0.5, conf=None, copy_paste=0.4, copy_paste_mode=flip, cos_lr=False, cutmix=0.2, data=data/cots_yolo_dataset/cots_yolo_dataset.yaml, degrees=90, deterministic=True, device=None, dfl=1.5, dnn=False, dropout=0.0, dynamic=False, embed=None, epochs=2, erasing=0.4, exist_ok=False, fliplr=0.5, flipud=0.5, format=torchscript, fraction=1.0, freeze=22, half=False, hsv_h=0.015, hsv_s=0.7, hsv_v=0.4, imgsz=1920, int8=False, iou=0.7, keras=False, kobj=1.0, line_width=None, lr0=0.0001, lrf=0.01, mask_ratio=4, max_det=300, mixup=0.2, mode=train, model=yolo11m.pt, momentum=0.937, mosaic=1.0, multi_scale=False, name=train17, nbs=64, nms=False, opset=None, optimize=False, optimizer=AdamW, overlap_mask=True, patience=5, perspective=0.0, plots=True, pose=12.0, pretrained=True, profile=False, project=None, rect=False, resume=False, retina_masks=False, save=True, save_conf=False, save_crop=False, save_dir=runs/detect/train17, save_frames=False, save_json=False, save_period=1, save_txt=False, scale=0.9, seed=0, shear=5, show=False, show_boxes=True, show_conf=True, show_labels=True, simplify=True, single_cls=False, source=None, split=val, stream_buffer=False, task=detect, time=None, tracker=botsort.yaml, translate=0.4, val=True, verbose=True, vid_stride=1, visualize=False, warmup_bias_lr=0.1, warmup_epochs=3.0, warmup_momentum=0.8, weight_decay=0.0005, workers=8, workspace=None\n",
+      "\u001b[34m\u001b[1mengine/trainer: \u001b[0magnostic_nms=False, amp=True, augment=False, auto_augment=randaugment, batch=3, bgr=0.0, box=7.5, cache=False, cfg=None, classes=None, close_mosaic=10, cls=0.5, conf=None, copy_paste=0.4, copy_paste_mode=flip, cos_lr=False, cutmix=0.2, data=data/cots_yolo_dataset/cots_yolo_dataset.yaml, degrees=90, deterministic=True, device=None, dfl=1.5, dnn=False, dropout=0.0, dynamic=False, embed=None, epochs=2, erasing=0.4, exist_ok=False, fliplr=0.5, flipud=0.5, format=torchscript, fraction=1.0, freeze=22, half=False, hsv_h=0.015, hsv_s=0.7, hsv_v=0.4, imgsz=1920, int8=False, iou=0.7, keras=False, kobj=1.0, line_width=None, lr0=0.0001, lrf=0.01, mask_ratio=4, max_det=300, mixup=0.2, mode=train, model=yolo11m.pt, momentum=0.937, mosaic=1.0, multi_scale=False, name=train, nbs=64, nms=False, opset=None, optimize=False, optimizer=AdamW, overlap_mask=True, patience=5, perspective=0.0, plots=True, pose=12.0, pretrained=True, profile=False, project=None, rect=False, resume=False, retina_masks=False, save=True, save_conf=False, save_crop=False, save_dir=runs/detect/train, save_frames=False, save_json=False, save_period=1, save_txt=False, scale=0.9, seed=0, shear=5, show=False, show_boxes=True, show_conf=True, show_labels=True, simplify=True, single_cls=False, source=None, split=val, stream_buffer=False, task=detect, time=None, tracker=botsort.yaml, translate=0.4, val=True, verbose=True, vid_stride=1, visualize=False, warmup_bias_lr=0.1, warmup_epochs=3.0, warmup_momentum=0.8, weight_decay=0.0005, workers=8, workspace=None\n",
       "Overriding model.yaml nc=80 with nc=1\n",
       "\n",
       "                   from  n    params  module                                       arguments                     \n",
@@ -1212,7 +1203,7 @@
       "YOLO11m summary: 231 layers, 20,053,779 parameters, 20,053,763 gradients, 68.2 GFLOPs\n",
       "\n",
       "Transferred 643/649 items from pretrained weights\n",
-      "\u001b[34m\u001b[1mTensorBoard: \u001b[0mStart with 'tensorboard --logdir runs/detect/train17', view at http://localhost:6006/\n",
+      "\u001b[34m\u001b[1mTensorBoard: \u001b[0mStart with 'tensorboard --logdir runs/detect/train', view at http://localhost:6006/\n",
       "Freezing layer 'model.0.conv.weight'\n",
       "Freezing layer 'model.0.bn.weight'\n",
       "Freezing layer 'model.0.bn.bias'\n",
@@ -1453,14 +1444,14 @@
       "Freezing layer 'model.23.dfl.conv.weight'\n",
       "\u001b[34m\u001b[1mAMP: \u001b[0mrunning Automatic Mixed Precision (AMP) checks...\n",
       "\u001b[34m\u001b[1mAMP: \u001b[0mchecks passed ✅\n",
-      "\u001b[34m\u001b[1mtrain: \u001b[0mFast image access ✅ (ping: 0.0±0.0 ms, read: 2511.9±3205.9 MB/s, size: 2272.2 KB)\n"
+      "\u001b[34m\u001b[1mtrain: \u001b[0mFast image access ✅ (ping: 0.0±0.0 ms, read: 2606.2±3398.3 MB/s, size: 2272.2 KB)\n"
      ]
     },
     {
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "\u001b[34m\u001b[1mtrain: \u001b[0mScanning /home/light/personal_projects/vision_projects/notebooks/data/cot\u001b[0m"
+      "\u001b[34m\u001b[1mtrain: \u001b[0mScanning /home/light/personal_projects/cots_detection_project/data/cots_yolo_dataset/labels/train.cache... 5235 im\u001b[0m"
      ]
     },
     {
@@ -1485,26 +1476,26 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "\u001b[34m\u001b[1mval: \u001b[0mFast image access ✅ (ping: 0.0±0.0 ms, read: 865.5±149.9 MB/s, size: 2282.0 KB)\n"
+      "\u001b[34m\u001b[1mval: \u001b[0mFast image access ✅ (ping: 0.0±0.0 ms, read: 657.0±143.6 MB/s, size: 2282.0 KB)\n"
      ]
     },
     {
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "\u001b[34m\u001b[1mval: \u001b[0mScanning /home/light/personal_projects/vision_projects/notebooks/data/cots_\u001b[0m\n"
+      "\u001b[34m\u001b[1mval: \u001b[0mScanning /home/light/personal_projects/cots_detection_project/data/cots_yolo_dataset/labels/val.cache... 917 images,\u001b[0m\n"
      ]
     },
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Plotting labels to runs/detect/train17/labels.jpg... \n",
+      "Plotting labels to runs/detect/train/labels.jpg... \n",
       "\u001b[34m\u001b[1moptimizer:\u001b[0m AdamW(lr=0.0001, momentum=0.937) with parameter groups 106 weight(decay=0.0), 113 weight(decay=0.0004921875), 112 bias(decay=0.0)\n",
       "\u001b[34m\u001b[1mTensorBoard: \u001b[0mmodel graph visualization added ✅\n",
       "Image sizes 1920 train, 1920 val\n",
       "Using 8 dataloader workers\n",
-      "Logging results to \u001b[1mruns/detect/train17\u001b[0m\n",
+      "Logging results to \u001b[1mruns/detect/train\u001b[0m\n",
       "Starting training for 2 epochs...\n",
       "\n",
       "      Epoch    GPU_mem   box_loss   cls_loss   dfl_loss  Instances       Size\n"
@@ -1514,8 +1505,8 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "        1/2      4.68G      2.198      3.821      2.373          4       1920: 1\n",
-      "                 Class     Images  Instances      Box(P          R      mAP50  m"
+      "        1/2      4.68G      2.198      3.821      2.373          4       1920: 100%|██████████| 1745/1745 [04:41<00:00,  \n",
+      "                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 153/153 [00:22"
      ]
     },
     {
@@ -1544,8 +1535,8 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "        2/2      4.68G          2      3.393      2.186          8       1920: 1\n",
-      "                 Class     Images  Instances      Box(P          R      mAP50  m\n"
+      "        2/2      4.68G          2      3.393      2.186          8       1920: 100%|██████████| 1745/1745 [04:52<00:00,  \n",
+      "                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 153/153 [00:21\n"
      ]
     },
     {
@@ -1554,11 +1545,11 @@
      "text": [
       "                   all        917       1768      0.358      0.128      0.117     0.0519\n",
       "\n",
-      "2 epochs completed in 0.177 hours.\n",
-      "Optimizer stripped from runs/detect/train17/weights/last.pt, 40.7MB\n",
-      "Optimizer stripped from runs/detect/train17/weights/best.pt, 40.7MB\n",
+      "2 epochs completed in 0.173 hours.\n",
+      "Optimizer stripped from runs/detect/train/weights/last.pt, 40.7MB\n",
+      "Optimizer stripped from runs/detect/train/weights/best.pt, 40.7MB\n",
       "\n",
-      "Validating runs/detect/train17/weights/best.pt...\n",
+      "Validating runs/detect/train/weights/best.pt...\n",
       "Ultralytics 8.3.161 🚀 Python-3.9.22 torch-2.7.1+cu126 CUDA:0 (NVIDIA GeForce RTX 4090 Laptop GPU, 15955MiB)\n",
       "YOLO11m summary (fused): 125 layers, 20,030,803 parameters, 0 gradients, 67.6 GFLOPs\n"
      ]
@@ -1567,7 +1558,7 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "                 Class     Images  Instances      Box(P          R      mAP50  m\n"
+      "                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 153/153 [00:19\n"
      ]
     },
     {
@@ -1575,11 +1566,11 @@
      "output_type": "stream",
      "text": [
       "                   all        917       1768      0.356      0.129      0.118      0.052\n",
-      "Speed: 0.8ms preprocess, 17.8ms inference, 0.0ms loss, 1.2ms postprocess per image\n",
-      "Results saved to \u001b[1mruns/detect/train17\u001b[0m\n",
+      "Speed: 0.7ms preprocess, 17.9ms inference, 0.0ms loss, 1.0ms postprocess per image\n",
+      "Results saved to \u001b[1mruns/detect/train\u001b[0m\n",
       "New https://pypi.org/project/ultralytics/8.3.176 available 😃 Update with 'pip install -U ultralytics'\n",
       "Ultralytics 8.3.161 🚀 Python-3.9.22 torch-2.7.1+cu126 CUDA:0 (NVIDIA GeForce RTX 4090 Laptop GPU, 15955MiB)\n",
-      "\u001b[34m\u001b[1mengine/trainer: \u001b[0magnostic_nms=False, amp=True, augment=False, auto_augment=randaugment, batch=3, bgr=0.0, box=7.5, cache=False, cfg=None, classes=None, close_mosaic=10, cls=0.5, conf=None, copy_paste=0.4, copy_paste_mode=flip, cos_lr=False, cutmix=0.2, data=data/cots_yolo_dataset/cots_yolo_dataset.yaml, degrees=90, deterministic=True, device=None, dfl=1.5, dnn=False, dropout=0.2, dynamic=False, embed=None, epochs=60, erasing=0.4, exist_ok=False, fliplr=0.5, flipud=0.5, format=torchscript, fraction=1.0, freeze=None, half=False, hsv_h=0.015, hsv_s=0.7, hsv_v=0.4, imgsz=1920, int8=False, iou=0.7, keras=False, kobj=1.0, line_width=None, lr0=0.0001, lrf=0.01, mask_ratio=4, max_det=300, mixup=0.2, mode=train, model=yolo11m.pt, momentum=0.937, mosaic=1.0, multi_scale=False, name=train172, nbs=64, nms=False, opset=None, optimize=False, optimizer=AdamW, overlap_mask=True, patience=5, perspective=0.0, plots=True, pose=12.0, pretrained=True, profile=False, project=None, rect=False, resume=False, retina_masks=False, save=True, save_conf=False, save_crop=False, save_dir=runs/detect/train172, save_frames=False, save_json=False, save_period=5, save_txt=False, scale=0.9, seed=0, shear=5, show=False, show_boxes=True, show_conf=True, show_labels=True, simplify=True, single_cls=False, source=None, split=val, stream_buffer=False, task=detect, time=None, tracker=botsort.yaml, translate=0.4, val=True, verbose=True, vid_stride=1, visualize=False, warmup_bias_lr=0.1, warmup_epochs=3.0, warmup_momentum=0.8, weight_decay=0.0005, workers=8, workspace=None\n",
+      "\u001b[34m\u001b[1mengine/trainer: \u001b[0magnostic_nms=False, amp=True, augment=False, auto_augment=randaugment, batch=3, bgr=0.0, box=7.5, cache=False, cfg=None, classes=None, close_mosaic=10, cls=0.5, conf=None, copy_paste=0.4, copy_paste_mode=flip, cos_lr=False, cutmix=0.2, data=data/cots_yolo_dataset/cots_yolo_dataset.yaml, degrees=90, deterministic=True, device=None, dfl=1.5, dnn=False, dropout=0.2, dynamic=False, embed=None, epochs=60, erasing=0.4, exist_ok=False, fliplr=0.5, flipud=0.5, format=torchscript, fraction=1.0, freeze=None, half=False, hsv_h=0.015, hsv_s=0.7, hsv_v=0.4, imgsz=1920, int8=False, iou=0.7, keras=False, kobj=1.0, line_width=None, lr0=0.0001, lrf=0.01, mask_ratio=4, max_det=300, mixup=0.2, mode=train, model=yolo11m.pt, momentum=0.937, mosaic=1.0, multi_scale=False, name=train2, nbs=64, nms=False, opset=None, optimize=False, optimizer=AdamW, overlap_mask=True, patience=5, perspective=0.0, plots=True, pose=12.0, pretrained=True, profile=False, project=None, rect=False, resume=False, retina_masks=False, save=True, save_conf=False, save_crop=False, save_dir=runs/detect/train2, save_frames=False, save_json=False, save_period=5, save_txt=False, scale=0.9, seed=0, shear=5, show=False, show_boxes=True, show_conf=True, show_labels=True, simplify=True, single_cls=False, source=None, split=val, stream_buffer=False, task=detect, time=None, tracker=botsort.yaml, translate=0.4, val=True, verbose=True, vid_stride=1, visualize=False, warmup_bias_lr=0.1, warmup_epochs=3.0, warmup_momentum=0.8, weight_decay=0.0005, workers=8, workspace=None\n",
       "\n",
       "                   from  n    params  module                                       arguments                     \n",
       "  0                  -1  1      1856  ultralytics.nn.modules.conv.Conv             [3, 64, 3, 2]                 \n",
@@ -1609,18 +1600,18 @@
       "YOLO11m summary: 231 layers, 20,053,779 parameters, 20,053,763 gradients, 68.2 GFLOPs\n",
       "\n",
       "Transferred 649/649 items from pretrained weights\n",
-      "\u001b[34m\u001b[1mTensorBoard: \u001b[0mStart with 'tensorboard --logdir runs/detect/train172', view at http://localhost:6006/\n",
+      "\u001b[34m\u001b[1mTensorBoard: \u001b[0mStart with 'tensorboard --logdir runs/detect/train2', view at http://localhost:6006/\n",
       "Freezing layer 'model.23.dfl.conv.weight'\n",
       "\u001b[34m\u001b[1mAMP: \u001b[0mrunning Automatic Mixed Precision (AMP) checks...\n",
       "\u001b[34m\u001b[1mAMP: \u001b[0mchecks passed ✅\n",
-      "\u001b[34m\u001b[1mtrain: \u001b[0mFast image access ✅ (ping: 0.0±0.0 ms, read: 3491.9±1668.3 MB/s, size: 2272.2 KB)\n"
+      "\u001b[34m\u001b[1mtrain: \u001b[0mFast image access ✅ (ping: 0.0±0.0 ms, read: 1639.9±1970.9 MB/s, size: 2272.2 KB)\n"
      ]
     },
     {
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "\u001b[34m\u001b[1mtrain: \u001b[0mScanning /home/light/personal_projects/vision_projects/notebooks/data/cot\u001b[0m"
+      "\u001b[34m\u001b[1mtrain: \u001b[0mScanning /home/light/personal_projects/cots_detection_project/data/cots_yolo_dataset/labels/train.cache... 5235 im\u001b[0m"
      ]
     },
     {
@@ -1645,26 +1636,26 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "\u001b[34m\u001b[1mval: \u001b[0mFast image access ✅ (ping: 0.0±0.0 ms, read: 3211.5±2298.3 MB/s, size: 2282.0 KB)\n"
+      "\u001b[34m\u001b[1mval: \u001b[0mFast image access ✅ (ping: 0.0±0.0 ms, read: 3665.5±2484.2 MB/s, size: 2282.0 KB)\n"
      ]
     },
     {
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "\u001b[34m\u001b[1mval: \u001b[0mScanning /home/light/personal_projects/vision_projects/notebooks/data/cots_\u001b[0m\n"
+      "\u001b[34m\u001b[1mval: \u001b[0mScanning /home/light/personal_projects/cots_detection_project/data/cots_yolo_dataset/labels/val.cache... 917 images,\u001b[0m\n"
      ]
     },
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Plotting labels to runs/detect/train172/labels.jpg... \n",
+      "Plotting labels to runs/detect/train2/labels.jpg... \n",
       "\u001b[34m\u001b[1moptimizer:\u001b[0m AdamW(lr=0.0001, momentum=0.937) with parameter groups 106 weight(decay=0.0), 113 weight(decay=0.0004921875), 112 bias(decay=0.0)\n",
       "\u001b[34m\u001b[1mTensorBoard: \u001b[0mmodel graph visualization added ✅\n",
       "Image sizes 1920 train, 1920 val\n",
       "Using 8 dataloader workers\n",
-      "Logging results to \u001b[1mruns/detect/train172\u001b[0m\n",
+      "Logging results to \u001b[1mruns/detect/train2\u001b[0m\n",
       "Starting training for 60 epochs...\n",
       "\n",
       "      Epoch    GPU_mem   box_loss   cls_loss   dfl_loss  Instances       Size\n"
@@ -1674,15 +1665,28 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "       1/60      13.5G      1.794      2.824      1.924          4       1920: 1\n",
-      "                 Class     Images  Instances      Box(P          R      mAP50  m\n"
+      "       1/60      13.5G      1.794      2.824      1.924          4       1920: 100%|██████████| 1745/1745 [10:43<00:00,  \n",
+      "                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 153/153 [00:20"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "                   all        917       1768      0.438      0.213      0.216     0.0882\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "\n"
      ]
     },
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "                   all        917       1768      0.438      0.213      0.216     0.0882\n",
       "\n",
       "      Epoch    GPU_mem   box_loss   cls_loss   dfl_loss  Instances       Size\n"
      ]
@@ -1691,8 +1695,8 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "       2/60      14.7G      1.569      2.243      1.706          8       1920: 1\n",
-      "                 Class     Images  Instances      Box(P          R      mAP50  m"
+      "       2/60      14.7G      1.569      2.243      1.706          8       1920: 100%|██████████| 1745/1745 [10:37<00:00,  \n",
+      "                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 153/153 [00:19"
      ]
     },
     {
@@ -1721,8 +1725,8 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "       3/60      14.7G      1.533       2.02      1.655         13       1920: 1\n",
-      "                 Class     Images  Instances      Box(P          R      mAP50  m"
+      "       3/60      14.7G      1.533       2.02      1.655         13       1920: 100%|██████████| 1745/1745 [10:37<00:00,  \n",
+      "                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 153/153 [00:20"
      ]
     },
     {
@@ -1751,8 +1755,8 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "       4/60      14.7G      1.457      1.888      1.606          7       1920: 1\n",
-      "                 Class     Images  Instances      Box(P          R      mAP50  m"
+      "       4/60      14.7G      1.457      1.888      1.606          7       1920: 100%|██████████| 1745/1745 [10:45<00:00,  \n",
+      "                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 153/153 [00:20"
      ]
     },
     {
@@ -1781,8 +1785,8 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "       5/60      14.7G      1.436      1.806       1.59         11       1920: 1\n",
-      "                 Class     Images  Instances      Box(P          R      mAP50  m"
+      "       5/60      14.7G      1.436      1.806       1.59         11       1920: 100%|██████████| 1745/1745 [10:38<00:00,  \n",
+      "                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 153/153 [00:21"
      ]
     },
     {
@@ -1811,8 +1815,8 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "       6/60      14.7G      1.419      1.736      1.576          5       1920: 1\n",
-      "                 Class     Images  Instances      Box(P          R      mAP50  m"
+      "       6/60      14.7G      1.419      1.736      1.576          5       1920: 100%|██████████| 1745/1745 [10:36<00:00,  \n",
+      "                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 153/153 [00:20"
      ]
     },
     {
@@ -1841,8 +1845,8 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "       7/60      14.7G      1.407      1.672      1.577         10       1920: 1\n",
-      "                 Class     Images  Instances      Box(P          R      mAP50  m"
+      "       7/60      14.7G      1.407      1.672      1.577         10       1920: 100%|██████████| 1745/1745 [10:45<00:00,  \n",
+      "                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 153/153 [00:20"
      ]
     },
     {
@@ -1871,8 +1875,8 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "       8/60      14.7G      1.403       1.63       1.56          9       1920: 1\n",
-      "                 Class     Images  Instances      Box(P          R      mAP50  m"
+      "       8/60      14.7G      1.403       1.63       1.56          9       1920: 100%|██████████| 1745/1745 [10:37<00:00,  \n",
+      "                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 153/153 [00:19"
      ]
     },
     {
@@ -1901,8 +1905,8 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "       9/60      14.7G      1.379      1.613      1.556          8       1920: 1\n",
-      "                 Class     Images  Instances      Box(P          R      mAP50  m"
+      "       9/60      14.7G      1.379      1.613      1.556          8       1920: 100%|██████████| 1745/1745 [10:41<00:00,  \n",
+      "                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 153/153 [00:19"
      ]
     },
     {
@@ -1931,8 +1935,8 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "      10/60      14.7G      1.375      1.562       1.54         10       1920: 1\n",
-      "                 Class     Images  Instances      Box(P          R      mAP50  m"
+      "      10/60      14.7G      1.375      1.562       1.54         10       1920: 100%|██████████| 1745/1745 [10:39<00:00,  \n",
+      "                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 153/153 [00:20"
      ]
     },
     {
@@ -1961,8 +1965,8 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "      11/60      14.7G      1.358      1.537      1.535         14       1920: 1\n",
-      "                 Class     Images  Instances      Box(P          R      mAP50  m"
+      "      11/60      14.7G      1.358      1.537      1.535         14       1920: 100%|██████████| 1745/1745 [10:34<00:00,  \n",
+      "                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 153/153 [00:20"
      ]
     },
     {
@@ -1991,8 +1995,8 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "      12/60      14.7G      1.353      1.492       1.54          8       1920: 1\n",
-      "                 Class     Images  Instances      Box(P          R      mAP50  m"
+      "      12/60      14.7G      1.353      1.492       1.54          8       1920: 100%|██████████| 1745/1745 [10:35<00:00,  \n",
+      "                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 153/153 [00:20"
      ]
     },
     {
@@ -2021,8 +2025,8 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "      13/60      14.7G      1.352      1.487      1.543         13       1920: 1\n",
-      "                 Class     Images  Instances      Box(P          R      mAP50  m"
+      "      13/60      14.7G      1.352      1.487      1.543         13       1920: 100%|██████████| 1745/1745 [10:41<00:00,  \n",
+      "                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 153/153 [00:21"
      ]
     },
     {
@@ -2051,8 +2055,8 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "      14/60      14.7G      1.345      1.498      1.533          6       1920: 1\n",
-      "                 Class     Images  Instances      Box(P          R      mAP50  m"
+      "      14/60      14.7G      1.345      1.498      1.533          6       1920: 100%|██████████| 1745/1745 [10:37<00:00,  \n",
+      "                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 153/153 [00:20"
      ]
     },
     {
@@ -2081,8 +2085,8 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "      15/60      14.7G      1.337       1.45      1.523          6       1920: 1\n",
-      "                 Class     Images  Instances      Box(P          R      mAP50  m"
+      "      15/60      14.7G      1.337       1.45      1.523          6       1920: 100%|██████████| 1745/1745 [10:36<00:00,  \n",
+      "                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 153/153 [00:20"
      ]
     },
     {
@@ -2106,11 +2110,11 @@
      "output_type": "stream",
      "text": [
       "\n",
-      "15 epochs completed in 2.839 hours.\n",
-      "Optimizer stripped from runs/detect/train172/weights/last.pt, 40.7MB\n",
-      "Optimizer stripped from runs/detect/train172/weights/best.pt, 40.7MB\n",
+      "15 epochs completed in 2.755 hours.\n",
+      "Optimizer stripped from runs/detect/train2/weights/last.pt, 40.7MB\n",
+      "Optimizer stripped from runs/detect/train2/weights/best.pt, 40.7MB\n",
       "\n",
-      "Validating runs/detect/train172/weights/best.pt...\n",
+      "Validating runs/detect/train2/weights/best.pt...\n",
       "Ultralytics 8.3.161 🚀 Python-3.9.22 torch-2.7.1+cu126 CUDA:0 (NVIDIA GeForce RTX 4090 Laptop GPU, 15955MiB)\n",
       "YOLO11m summary (fused): 125 layers, 20,030,803 parameters, 0 gradients, 67.6 GFLOPs\n"
      ]
@@ -2119,7 +2123,7 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "                 Class     Images  Instances      Box(P          R      mAP50  m\n"
+      "                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 153/153 [00:20\n"
      ]
     },
     {
@@ -2127,14 +2131,19 @@
      "output_type": "stream",
      "text": [
       "                   all        917       1768      0.817      0.637      0.702      0.405\n",
-      "Speed: 0.8ms preprocess, 17.9ms inference, 0.0ms loss, 1.0ms postprocess per image\n",
-      "Results saved to \u001b[1mruns/detect/train172\u001b[0m\n"
+      "Speed: 0.8ms preprocess, 17.6ms inference, 0.0ms loss, 1.3ms postprocess per image\n",
+      "Results saved to \u001b[1mruns/detect/train2\u001b[0m\n"
      ]
     }
    ],
    "source": [
     "from ultralytics import YOLO\n",
     "\n",
+    "#Download pretrained model\n",
+    "import urllib.request\n",
+    "url = 'https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11m.pt'\n",
+    "urllib.request.urlretrieve(url, 'yolo11m.pt')\n",
+    "\n",
     "# Load a pretrain model\n",
     "model = YOLO('yolo11m.pt')\n",
     "\n",
@@ -2173,7 +2182,8 @@
     "  task='detect',\n",
     "  save_period=1,\n",
     "  freeze = 22, #Freeze the first layers\n",
-    "  **data_augment_args\n",
+    "  **data_augment_args,\n",
+    "  **training_args\n",
     ")\n",
     "\n",
     "# Fine tune all model\n",
@@ -2186,6 +2196,7 @@
     "  dropout=0.2,\n",
     "  freeze = None,\n",
     "  **data_augment_args,\n",
+    "  **training_args\n",
     ")\n",
     "\n",
     "                     "
@@ -2201,7 +2212,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 29,
+   "execution_count": 28,
    "id": "459324cd-a36f-4d81-90f6-779ad3b7aff6",
    "metadata": {},
    "outputs": [],
@@ -2211,7 +2222,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 30,
+   "execution_count": 29,
    "id": "58487efb-b7db-44ed-8eda-c903b06f45bb",
    "metadata": {},
    "outputs": [
@@ -2221,26 +2232,26 @@
      "text": [
       "Requirement already satisfied: supervision in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (0.25.1)\n",
       "Requirement already satisfied: opencv-python>=4.5.5.64 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from supervision) (4.11.0.86)\n",
-      "Requirement already satisfied: pyyaml>=5.3 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from supervision) (6.0.2)\n",
-      "Requirement already satisfied: requests>=2.26.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from supervision) (2.32.4)\n",
-      "Requirement already satisfied: tqdm>=4.62.3 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from supervision) (4.67.1)\n",
-      "Requirement already satisfied: scipy<2.0.0,>=1.10.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from supervision) (1.13.1)\n",
+      "Requirement already satisfied: matplotlib>=3.6.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from supervision) (3.9.4)\n",
       "Requirement already satisfied: contourpy>=1.0.7 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from supervision) (1.3.0)\n",
       "Requirement already satisfied: defusedxml<0.8.0,>=0.7.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from supervision) (0.7.1)\n",
-      "Requirement already satisfied: pillow>=9.4 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from supervision) (10.4.0)\n",
+      "Requirement already satisfied: tqdm>=4.62.3 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from supervision) (4.67.1)\n",
+      "Requirement already satisfied: requests>=2.26.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from supervision) (2.32.4)\n",
+      "Requirement already satisfied: scipy<2.0.0,>=1.10.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from supervision) (1.13.1)\n",
       "Requirement already satisfied: numpy>=1.21.2 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from supervision) (2.0.2)\n",
-      "Requirement already satisfied: matplotlib>=3.6.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from supervision) (3.9.4)\n",
-      "Requirement already satisfied: kiwisolver>=1.3.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.6.0->supervision) (1.4.7)\n",
+      "Requirement already satisfied: pyyaml>=5.3 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from supervision) (6.0.2)\n",
+      "Requirement already satisfied: pillow>=9.4 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from supervision) (10.4.0)\n",
       "Requirement already satisfied: python-dateutil>=2.7 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.6.0->supervision) (2.9.0.post0)\n",
-      "Requirement already satisfied: cycler>=0.10 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.6.0->supervision) (0.12.1)\n",
+      "Requirement already satisfied: fonttools>=4.22.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.6.0->supervision) (4.58.4)\n",
       "Requirement already satisfied: importlib-resources>=3.2.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.6.0->supervision) (6.5.2)\n",
+      "Requirement already satisfied: kiwisolver>=1.3.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.6.0->supervision) (1.4.7)\n",
+      "Requirement already satisfied: cycler>=0.10 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.6.0->supervision) (0.12.1)\n",
       "Requirement already satisfied: packaging>=20.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.6.0->supervision) (25.0)\n",
-      "Requirement already satisfied: fonttools>=4.22.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.6.0->supervision) (4.58.4)\n",
       "Requirement already satisfied: pyparsing>=2.3.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from matplotlib>=3.6.0->supervision) (3.2.3)\n",
       "Requirement already satisfied: idna<4,>=2.5 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from requests>=2.26.0->supervision) (3.10)\n",
-      "Requirement already satisfied: charset_normalizer<4,>=2 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from requests>=2.26.0->supervision) (3.4.2)\n",
       "Requirement already satisfied: certifi>=2017.4.17 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from requests>=2.26.0->supervision) (2025.6.15)\n",
       "Requirement already satisfied: urllib3<3,>=1.21.1 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from requests>=2.26.0->supervision) (2.5.0)\n",
+      "Requirement already satisfied: charset_normalizer<4,>=2 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from requests>=2.26.0->supervision) (3.4.2)\n",
       "Requirement already satisfied: zipp>=3.1.0 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from importlib-resources>=3.2.0->matplotlib>=3.6.0->supervision) (3.23.0)\n",
       "Requirement already satisfied: six>=1.5 in /home/light/python_envs/ultralytics/lib/python3.9/site-packages (from python-dateutil>=2.7->matplotlib>=3.6.0->supervision) (1.17.0)\n",
       "\n",
@@ -2255,7 +2266,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 31,
+   "execution_count": 30,
    "id": "a8f773c2-b1e8-4853-8b7f-99f4b9a59c74",
    "metadata": {},
    "outputs": [],
@@ -2278,7 +2289,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 32,
+   "execution_count": 31,
    "id": "aaecd1ac-0d53-4aa3-a278-e3bc3be554c0",
    "metadata": {},
    "outputs": [],
@@ -2292,7 +2303,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 33,
+   "execution_count": 32,
    "id": "537c83a3-2a37-43b9-8a2f-09a76ea4634b",
    "metadata": {},
    "outputs": [],
@@ -2333,7 +2344,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 34,
+   "execution_count": 33,
    "id": "312ae47e-c77d-4c92-80ae-775a5e1b34b6",
    "metadata": {},
    "outputs": [
@@ -2364,7 +2375,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 35,
+   "execution_count": 34,
    "id": "d9d5dba7-3023-49c9-8719-de8d46c94678",
    "metadata": {},
    "outputs": [
@@ -2395,7 +2406,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 36,
+   "execution_count": 35,
    "id": "b6dae939-27fb-49f9-9839-7d446aded44d",
    "metadata": {},
    "outputs": [
@@ -2446,7 +2457,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 37,
+   "execution_count": 36,
    "id": "449a20dc-79cd-4b78-add2-a97bedfea375",
    "metadata": {},
    "outputs": [
@@ -2455,15 +2466,15 @@
      "output_type": "stream",
      "text": [
       "Ultralytics 8.3.161 🚀 Python-3.9.22 torch-2.7.1+cu126 CUDA:0 (NVIDIA GeForce RTX 4090 Laptop GPU, 15955MiB)\n",
-      "\u001b[34m\u001b[1mval: \u001b[0mFast image access ✅ (ping: 0.0±0.0 ms, read: 977.7±94.8 MB/s, size: 2668.4 KB)\n"
+      "\u001b[34m\u001b[1mval: \u001b[0mFast image access ✅ (ping: 0.0±0.0 ms, read: 668.0±301.8 MB/s, size: 2668.4 KB)\n"
      ]
     },
     {
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "\u001b[34m\u001b[1mval: \u001b[0mScanning /home/light/personal_projects/vision_projects/notebooks/data/cots_\u001b[0m\n",
-      "                 Class     Images  Instances      Box(P          R      mAP50  m\n"
+      "\u001b[34m\u001b[1mval: \u001b[0mScanning /home/light/personal_projects/cots_detection_project/data/cots_yolo_dataset/labels/train.cache... 5235 imag\u001b[0m\n",
+      "                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 2618/2618 [03:\n"
      ]
     },
     {
@@ -2471,8 +2482,8 @@
      "output_type": "stream",
      "text": [
       "                   all       5235       6436      0.887      0.833        0.9      0.574\n",
-      "Speed: 0.8ms preprocess, 33.5ms inference, 0.0ms loss, 0.9ms postprocess per image\n",
-      "Results saved to \u001b[1mruns/detect/train1722\u001b[0m\n",
+      "Speed: 0.8ms preprocess, 33.9ms inference, 0.0ms loss, 1.0ms postprocess per image\n",
+      "Results saved to \u001b[1mruns/detect/train22\u001b[0m\n",
       "F2 Score:  0.8435971699816329\n",
       "mAP50  0.8997627559970964\n",
       "mAP50-95  0.5735211022613346\n"
@@ -2493,7 +2504,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 38,
+   "execution_count": 37,
    "id": "a2260e31-b364-42f1-b410-77376a7f5941",
    "metadata": {},
    "outputs": [
@@ -2502,15 +2513,15 @@
      "output_type": "stream",
      "text": [
       "Ultralytics 8.3.161 🚀 Python-3.9.22 torch-2.7.1+cu126 CUDA:0 (NVIDIA GeForce RTX 4090 Laptop GPU, 15955MiB)\n",
-      "\u001b[34m\u001b[1mval: \u001b[0mFast image access ✅ (ping: 0.0±0.0 ms, read: 1501.4±1399.0 MB/s, size: 2504.4 KB)\n"
+      "\u001b[34m\u001b[1mval: \u001b[0mFast image access ✅ (ping: 0.0±0.0 ms, read: 765.4±135.0 MB/s, size: 2504.4 KB)\n"
      ]
     },
     {
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "\u001b[34m\u001b[1mval: \u001b[0mScanning /home/light/personal_projects/vision_projects/notebooks/data/cots_\u001b[0m\n",
-      "                 Class     Images  Instances      Box(P          R      mAP50  m\n"
+      "\u001b[34m\u001b[1mval: \u001b[0mScanning /home/light/personal_projects/cots_detection_project/data/cots_yolo_dataset/labels/val.cache... 917 images,\u001b[0m\n",
+      "                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 459/459 [00:34\n"
      ]
     },
     {
@@ -2518,8 +2529,8 @@
      "output_type": "stream",
      "text": [
       "                   all        917       1768      0.821      0.637      0.702      0.407\n",
-      "Speed: 1.0ms preprocess, 34.3ms inference, 0.0ms loss, 1.1ms postprocess per image\n",
-      "Results saved to \u001b[1mruns/detect/train1723\u001b[0m\n",
+      "Speed: 0.8ms preprocess, 33.8ms inference, 0.0ms loss, 0.7ms postprocess per image\n",
+      "Results saved to \u001b[1mruns/detect/train23\u001b[0m\n",
       "F2 Score:  0.6665938566597774\n",
       "mAP50  0.7020953206267737\n",
       "mAP50-95  0.40690137026758155\n"
@@ -2540,7 +2551,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 39,
+   "execution_count": 38,
    "id": "65a5d850-ccf2-4d5b-99b3-b944751b84d1",
    "metadata": {},
    "outputs": [
@@ -2549,15 +2560,29 @@
      "output_type": "stream",
      "text": [
       "Ultralytics 8.3.161 🚀 Python-3.9.22 torch-2.7.1+cu126 CUDA:0 (NVIDIA GeForce RTX 4090 Laptop GPU, 15955MiB)\n",
-      "\u001b[34m\u001b[1mval: \u001b[0mFast image access ✅ (ping: 0.0±0.0 ms, read: 638.3±78.6 MB/s, size: 2563.2 KB)\n"
+      "\u001b[34m\u001b[1mval: \u001b[0mFast image access ✅ (ping: 0.0±0.0 ms, read: 642.7±87.8 MB/s, size: 2563.2 KB)\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "\u001b[34m\u001b[1mval: \u001b[0mScanning /home/light/personal_projects/cots_detection_project/data/cots_yolo_dataset/labels/test... 918 images, 581 \u001b[0m"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\u001b[34m\u001b[1mval: \u001b[0mNew cache created: /home/light/personal_projects/cots_detection_project/data/cots_yolo_dataset/labels/test.cache\n"
      ]
     },
     {
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "\u001b[34m\u001b[1mval: \u001b[0mScanning /home/light/personal_projects/vision_projects/notebooks/data/cots_\u001b[0m\n",
-      "                 Class     Images  Instances      Box(P          R      mAP50  m\n"
+      "\n",
+      "                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 750/750 [00:57\n"
      ]
     },
     {
@@ -2565,8 +2590,8 @@
      "output_type": "stream",
      "text": [
       "                   all       1499       1004      0.877      0.764      0.819      0.377\n",
-      "Speed: 0.8ms preprocess, 33.7ms inference, 0.0ms loss, 0.8ms postprocess per image\n",
-      "Results saved to \u001b[1mruns/detect/train1724\u001b[0m\n",
+      "Speed: 0.9ms preprocess, 33.6ms inference, 0.0ms loss, 0.9ms postprocess per image\n",
+      "Results saved to \u001b[1mruns/detect/train24\u001b[0m\n",
       "F2 Score:  0.7841299670635844\n",
       "mAP50  0.8192513111358107\n",
       "mAP50-95  0.37744414058421033\n"