{ "cells": [ { "cell_type": "code", "execution_count": 1, "metadata": { "colab": { "base_uri": "https://localhost:8080/" }, "id": "Mw7RoF7eRJEG", "outputId": "1aa46197-e525-4227-b78a-5068873ebd65" }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Cloning into 'YOLO9_KMUTT'...\n", "remote: Enumerating objects: 182, done.\u001b[K\n", "remote: Counting objects: 100% (182/182), done.\u001b[K\n", "remote: Compressing objects: 100% (132/132), done.\u001b[K\n", "remote: Total 182 (delta 49), reused 179 (delta 46), pack-reused 0\u001b[K\n", "Receiving objects: 100% (182/182), 2.27 MiB | 8.47 MiB/s, done.\n", "Resolving deltas: 100% (49/49), done.\n" ] } ], "source": [ "!git clone https://github.com/Sompote/YOLO9_KMUTT" ] }, { "cell_type": "code", "execution_count": 9, "metadata": { "colab": { "base_uri": "https://localhost:8080/" }, "id": "6KgJYQKfryz2", "outputId": "f359b42d-f9d5-41ac-bbc6-d4c623672e92" }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "/workspace/YOLO9_KMUTT/yolov9\n" ] } ], "source": [ "%cd /workspace/YOLO9_KMUTT/yolov9/" ] }, { "cell_type": "code", "execution_count": 10, "metadata": { "colab": { "base_uri": "https://localhost:8080/" }, "id": "AnohaOgqq5rQ", "outputId": "d4a78b77-33f3-43dd-af6e-c1ef5e32a8f9" }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Collecting gitpython (from -r requirements.txt (line 5))\n", " Downloading GitPython-3.1.43-py3-none-any.whl.metadata (13 kB)\n", "Requirement already satisfied: ipython in /opt/conda/lib/python3.10/site-packages (from -r requirements.txt (line 6)) (8.20.0)\n", "Collecting matplotlib>=3.2.2 (from -r requirements.txt (line 7))\n", " Downloading matplotlib-3.9.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (11 kB)\n", "Requirement already satisfied: numpy>=1.18.5 in /opt/conda/lib/python3.10/site-packages (from -r requirements.txt (line 8)) (1.26.3)\n", "Collecting opencv-python>=4.1.1 (from -r requirements.txt (line 9))\n", " Downloading opencv_python-4.10.0.82-cp37-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (20 kB)\n", "Requirement already satisfied: Pillow>=7.1.2 in /opt/conda/lib/python3.10/site-packages (from -r requirements.txt (line 10)) (10.0.1)\n", "Requirement already satisfied: psutil in /opt/conda/lib/python3.10/site-packages (from -r requirements.txt (line 11)) (5.9.0)\n", "Requirement already satisfied: PyYAML>=5.3.1 in /opt/conda/lib/python3.10/site-packages (from -r requirements.txt (line 12)) (6.0.1)\n", "Requirement already satisfied: requests>=2.23.0 in /opt/conda/lib/python3.10/site-packages (from -r requirements.txt (line 13)) (2.31.0)\n", "Collecting scipy>=1.4.1 (from -r requirements.txt (line 14))\n", " Downloading scipy-1.13.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (60 kB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m60.6/60.6 kB\u001b[0m \u001b[31m9.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", "\u001b[?25hCollecting thop>=0.1.1 (from -r requirements.txt (line 15))\n", " Downloading thop-0.1.1.post2209072238-py3-none-any.whl.metadata (2.7 kB)\n", "Requirement already satisfied: torch>=1.7.0 in /opt/conda/lib/python3.10/site-packages (from -r requirements.txt (line 16)) (2.2.0)\n", "Requirement already satisfied: torchvision>=0.8.1 in /opt/conda/lib/python3.10/site-packages (from -r requirements.txt (line 17)) (0.17.0)\n", "Requirement already satisfied: tqdm>=4.64.0 in /opt/conda/lib/python3.10/site-packages (from -r requirements.txt (line 18)) (4.65.0)\n", "Collecting tensorboard>=2.4.1 (from -r requirements.txt (line 22))\n", " Downloading tensorboard-2.17.0-py3-none-any.whl.metadata (1.6 kB)\n", "Collecting pandas>=1.1.4 (from -r requirements.txt (line 27))\n", " Downloading pandas-2.2.2-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (19 kB)\n", "Collecting seaborn>=0.11.0 (from -r requirements.txt (line 28))\n", " Downloading seaborn-0.13.2-py3-none-any.whl.metadata (5.4 kB)\n", "Collecting albumentations>=1.0.3 (from -r requirements.txt (line 46))\n", " Downloading albumentations-1.4.8-py3-none-any.whl.metadata (37 kB)\n", "Collecting pycocotools>=2.0 (from -r requirements.txt (line 47))\n", " Downloading pycocotools-2.0.7-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (1.1 kB)\n", "Collecting gitdb<5,>=4.0.1 (from gitpython->-r requirements.txt (line 5))\n", " Downloading gitdb-4.0.11-py3-none-any.whl.metadata (1.2 kB)\n", "Requirement already satisfied: decorator in /opt/conda/lib/python3.10/site-packages (from ipython->-r requirements.txt (line 6)) (5.1.1)\n", "Requirement already satisfied: jedi>=0.16 in /opt/conda/lib/python3.10/site-packages (from ipython->-r requirements.txt (line 6)) (0.18.1)\n", "Requirement already satisfied: matplotlib-inline in /opt/conda/lib/python3.10/site-packages (from ipython->-r requirements.txt (line 6)) (0.1.6)\n", "Requirement already satisfied: prompt-toolkit<3.1.0,>=3.0.41 in /opt/conda/lib/python3.10/site-packages (from ipython->-r requirements.txt (line 6)) (3.0.43)\n", "Requirement already satisfied: pygments>=2.4.0 in /opt/conda/lib/python3.10/site-packages (from ipython->-r requirements.txt (line 6)) (2.15.1)\n", "Requirement already satisfied: stack-data in /opt/conda/lib/python3.10/site-packages (from ipython->-r requirements.txt (line 6)) (0.2.0)\n", "Requirement already satisfied: traitlets>=5 in /opt/conda/lib/python3.10/site-packages (from ipython->-r requirements.txt (line 6)) (5.7.1)\n", "Requirement already satisfied: exceptiongroup in /opt/conda/lib/python3.10/site-packages (from ipython->-r requirements.txt (line 6)) (1.2.0)\n", "Requirement already satisfied: pexpect>4.3 in /opt/conda/lib/python3.10/site-packages (from ipython->-r requirements.txt (line 6)) (4.8.0)\n", "Collecting contourpy>=1.0.1 (from matplotlib>=3.2.2->-r requirements.txt (line 7))\n", " Downloading contourpy-1.2.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (5.8 kB)\n", "Collecting cycler>=0.10 (from matplotlib>=3.2.2->-r requirements.txt (line 7))\n", " Downloading cycler-0.12.1-py3-none-any.whl.metadata (3.8 kB)\n", "Collecting fonttools>=4.22.0 (from matplotlib>=3.2.2->-r requirements.txt (line 7))\n", " Downloading fonttools-4.53.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (162 kB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m162.2/162.2 kB\u001b[0m \u001b[31m22.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", "\u001b[?25hCollecting kiwisolver>=1.3.1 (from matplotlib>=3.2.2->-r requirements.txt (line 7))\n", " Downloading kiwisolver-1.4.5-cp310-cp310-manylinux_2_12_x86_64.manylinux2010_x86_64.whl.metadata (6.4 kB)\n", "Requirement already satisfied: packaging>=20.0 in /opt/conda/lib/python3.10/site-packages (from matplotlib>=3.2.2->-r requirements.txt (line 7)) (23.1)\n", "Collecting pyparsing>=2.3.1 (from matplotlib>=3.2.2->-r requirements.txt (line 7))\n", " Downloading pyparsing-3.1.2-py3-none-any.whl.metadata (5.1 kB)\n", "Requirement already satisfied: python-dateutil>=2.7 in /opt/conda/lib/python3.10/site-packages (from matplotlib>=3.2.2->-r requirements.txt (line 7)) (2.9.0.post0)\n", "Requirement already satisfied: charset-normalizer<4,>=2 in /opt/conda/lib/python3.10/site-packages (from requests>=2.23.0->-r requirements.txt (line 13)) (2.0.4)\n", "Requirement already satisfied: idna<4,>=2.5 in /opt/conda/lib/python3.10/site-packages (from requests>=2.23.0->-r requirements.txt (line 13)) (3.4)\n", "Requirement already satisfied: urllib3<3,>=1.21.1 in /opt/conda/lib/python3.10/site-packages (from requests>=2.23.0->-r requirements.txt (line 13)) (1.26.18)\n", "Requirement already satisfied: certifi>=2017.4.17 in /opt/conda/lib/python3.10/site-packages (from requests>=2.23.0->-r requirements.txt (line 13)) (2023.11.17)\n", "Requirement already satisfied: filelock in /opt/conda/lib/python3.10/site-packages (from torch>=1.7.0->-r requirements.txt (line 16)) (3.13.1)\n", "Requirement already satisfied: typing-extensions>=4.8.0 in /opt/conda/lib/python3.10/site-packages (from torch>=1.7.0->-r requirements.txt (line 16)) (4.9.0)\n", "Requirement already satisfied: sympy in /opt/conda/lib/python3.10/site-packages (from torch>=1.7.0->-r requirements.txt (line 16)) (1.12)\n", "Requirement already satisfied: networkx in /opt/conda/lib/python3.10/site-packages (from torch>=1.7.0->-r requirements.txt (line 16)) (3.1)\n", "Requirement already satisfied: jinja2 in /opt/conda/lib/python3.10/site-packages (from torch>=1.7.0->-r requirements.txt (line 16)) (3.1.2)\n", "Requirement already satisfied: fsspec in /opt/conda/lib/python3.10/site-packages (from torch>=1.7.0->-r requirements.txt (line 16)) (2023.12.2)\n", "Collecting absl-py>=0.4 (from tensorboard>=2.4.1->-r requirements.txt (line 22))\n", " Downloading absl_py-2.1.0-py3-none-any.whl.metadata (2.3 kB)\n", "Collecting grpcio>=1.48.2 (from tensorboard>=2.4.1->-r requirements.txt (line 22))\n", " Downloading grpcio-1.64.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (3.3 kB)\n", "Collecting markdown>=2.6.8 (from tensorboard>=2.4.1->-r requirements.txt (line 22))\n", " Downloading Markdown-3.6-py3-none-any.whl.metadata (7.0 kB)\n", "Collecting protobuf!=4.24.0,<5.0.0,>=3.19.6 (from tensorboard>=2.4.1->-r requirements.txt (line 22))\n", " Downloading protobuf-4.25.3-cp37-abi3-manylinux2014_x86_64.whl.metadata (541 bytes)\n", "Requirement already satisfied: setuptools>=41.0.0 in /opt/conda/lib/python3.10/site-packages (from tensorboard>=2.4.1->-r requirements.txt (line 22)) (68.2.2)\n", "Requirement already satisfied: six>1.9 in /opt/conda/lib/python3.10/site-packages (from tensorboard>=2.4.1->-r requirements.txt (line 22)) (1.16.0)\n", "Collecting tensorboard-data-server<0.8.0,>=0.7.0 (from tensorboard>=2.4.1->-r requirements.txt (line 22))\n", " Downloading tensorboard_data_server-0.7.2-py3-none-manylinux_2_31_x86_64.whl.metadata (1.1 kB)\n", "Collecting werkzeug>=1.0.1 (from tensorboard>=2.4.1->-r requirements.txt (line 22))\n", " Downloading werkzeug-3.0.3-py3-none-any.whl.metadata (3.7 kB)\n", "Requirement already satisfied: pytz>=2020.1 in /opt/conda/lib/python3.10/site-packages (from pandas>=1.1.4->-r requirements.txt (line 27)) (2023.3.post1)\n", "Collecting tzdata>=2022.7 (from pandas>=1.1.4->-r requirements.txt (line 27))\n", " Downloading tzdata-2024.1-py2.py3-none-any.whl.metadata (1.4 kB)\n", "Collecting scikit-image>=0.21.0 (from albumentations>=1.0.3->-r requirements.txt (line 46))\n", " Downloading scikit_image-0.23.2-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (14 kB)\n", "Collecting scikit-learn>=1.3.2 (from albumentations>=1.0.3->-r requirements.txt (line 46))\n", " Downloading scikit_learn-1.5.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (11 kB)\n", "Collecting pydantic>=2.7.0 (from albumentations>=1.0.3->-r requirements.txt (line 46))\n", " Downloading pydantic-2.7.3-py3-none-any.whl.metadata (108 kB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m109.0/109.0 kB\u001b[0m \u001b[31m31.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", "\u001b[?25hCollecting albucore>=0.0.4 (from albumentations>=1.0.3->-r requirements.txt (line 46))\n", " Downloading albucore-0.0.9-py3-none-any.whl.metadata (3.1 kB)\n", "Collecting opencv-python-headless>=4.9.0.80 (from albumentations>=1.0.3->-r requirements.txt (line 46))\n", " Downloading opencv_python_headless-4.10.0.82-cp37-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (20 kB)\n", "Requirement already satisfied: tomli>=2.0.1 in /opt/conda/lib/python3.10/site-packages (from albucore>=0.0.4->albumentations>=1.0.3->-r requirements.txt (line 46)) (2.0.1)\n", "Collecting smmap<6,>=3.0.1 (from gitdb<5,>=4.0.1->gitpython->-r requirements.txt (line 5))\n", " Downloading smmap-5.0.1-py3-none-any.whl.metadata (4.3 kB)\n", "Requirement already satisfied: parso<0.9.0,>=0.8.0 in /opt/conda/lib/python3.10/site-packages (from jedi>=0.16->ipython->-r requirements.txt (line 6)) (0.8.3)\n", "Requirement already satisfied: ptyprocess>=0.5 in /opt/conda/lib/python3.10/site-packages (from pexpect>4.3->ipython->-r requirements.txt (line 6)) (0.7.0)\n", "Requirement already satisfied: wcwidth in /opt/conda/lib/python3.10/site-packages (from prompt-toolkit<3.1.0,>=3.0.41->ipython->-r requirements.txt (line 6)) (0.2.5)\n", "Collecting annotated-types>=0.4.0 (from pydantic>=2.7.0->albumentations>=1.0.3->-r requirements.txt (line 46))\n", " Downloading annotated_types-0.7.0-py3-none-any.whl.metadata (15 kB)\n", "Collecting pydantic-core==2.18.4 (from pydantic>=2.7.0->albumentations>=1.0.3->-r requirements.txt (line 46))\n", " Downloading pydantic_core-2.18.4-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (6.5 kB)\n", "Collecting imageio>=2.33 (from scikit-image>=0.21.0->albumentations>=1.0.3->-r requirements.txt (line 46))\n", " Downloading imageio-2.34.1-py3-none-any.whl.metadata (4.9 kB)\n", "Collecting tifffile>=2022.8.12 (from scikit-image>=0.21.0->albumentations>=1.0.3->-r requirements.txt (line 46))\n", " Downloading tifffile-2024.5.22-py3-none-any.whl.metadata (30 kB)\n", "Collecting lazy-loader>=0.4 (from scikit-image>=0.21.0->albumentations>=1.0.3->-r requirements.txt (line 46))\n", " Downloading lazy_loader-0.4-py3-none-any.whl.metadata (7.6 kB)\n", "Collecting joblib>=1.2.0 (from scikit-learn>=1.3.2->albumentations>=1.0.3->-r requirements.txt (line 46))\n", " Downloading joblib-1.4.2-py3-none-any.whl.metadata (5.4 kB)\n", "Collecting threadpoolctl>=3.1.0 (from scikit-learn>=1.3.2->albumentations>=1.0.3->-r requirements.txt (line 46))\n", " Downloading threadpoolctl-3.5.0-py3-none-any.whl.metadata (13 kB)\n", "Requirement already satisfied: MarkupSafe>=2.1.1 in /opt/conda/lib/python3.10/site-packages (from werkzeug>=1.0.1->tensorboard>=2.4.1->-r requirements.txt (line 22)) (2.1.3)\n", "Requirement already satisfied: executing in /opt/conda/lib/python3.10/site-packages (from stack-data->ipython->-r requirements.txt (line 6)) (0.8.3)\n", "Requirement already satisfied: asttokens in /opt/conda/lib/python3.10/site-packages (from stack-data->ipython->-r requirements.txt (line 6)) (2.0.5)\n", "Requirement already satisfied: pure-eval in /opt/conda/lib/python3.10/site-packages (from stack-data->ipython->-r requirements.txt (line 6)) (0.2.2)\n", "Requirement already satisfied: mpmath>=0.19 in /opt/conda/lib/python3.10/site-packages (from sympy->torch>=1.7.0->-r requirements.txt (line 16)) (1.3.0)\n", "Downloading GitPython-3.1.43-py3-none-any.whl (207 kB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m207.3/207.3 kB\u001b[0m \u001b[31m35.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", "\u001b[?25hDownloading matplotlib-3.9.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (8.3 MB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m8.3/8.3 MB\u001b[0m \u001b[31m127.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0ma \u001b[36m0:00:01\u001b[0m\n", "\u001b[?25hDownloading opencv_python-4.10.0.82-cp37-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (62.5 MB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m62.5/62.5 MB\u001b[0m \u001b[31m89.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m:00:01\u001b[0m00:01\u001b[0m\n", "\u001b[?25hDownloading scipy-1.13.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (38.6 MB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m38.6/38.6 MB\u001b[0m \u001b[31m100.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m00:01\u001b[0m00:01\u001b[0m\n", "\u001b[?25hDownloading thop-0.1.1.post2209072238-py3-none-any.whl (15 kB)\n", "Downloading tensorboard-2.17.0-py3-none-any.whl (5.5 MB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m5.5/5.5 MB\u001b[0m \u001b[31m148.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m00:01\u001b[0m\n", "\u001b[?25hDownloading pandas-2.2.2-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (13.0 MB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m13.0/13.0 MB\u001b[0m \u001b[31m151.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m00:01\u001b[0m0:01\u001b[0m\n", "\u001b[?25hDownloading seaborn-0.13.2-py3-none-any.whl (294 kB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m294.9/294.9 kB\u001b[0m \u001b[31m50.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", "\u001b[?25hDownloading albumentations-1.4.8-py3-none-any.whl (156 kB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m156.8/156.8 kB\u001b[0m \u001b[31m35.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", "\u001b[?25hDownloading pycocotools-2.0.7-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (426 kB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m426.2/426.2 kB\u001b[0m \u001b[31m61.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", "\u001b[?25hDownloading absl_py-2.1.0-py3-none-any.whl (133 kB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m133.7/133.7 kB\u001b[0m \u001b[31m33.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", "\u001b[?25hDownloading albucore-0.0.9-py3-none-any.whl (7.9 kB)\n", "Downloading contourpy-1.2.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (305 kB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m305.2/305.2 kB\u001b[0m \u001b[31m50.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", "\u001b[?25hDownloading cycler-0.12.1-py3-none-any.whl (8.3 kB)\n", "Downloading fonttools-4.53.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (4.6 MB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m4.6/4.6 MB\u001b[0m \u001b[31m141.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", "\u001b[?25hDownloading gitdb-4.0.11-py3-none-any.whl (62 kB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m62.7/62.7 kB\u001b[0m \u001b[31m13.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", "\u001b[?25hDownloading grpcio-1.64.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (5.6 MB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m5.6/5.6 MB\u001b[0m \u001b[31m140.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m00:01\u001b[0m\n", "\u001b[?25hDownloading kiwisolver-1.4.5-cp310-cp310-manylinux_2_12_x86_64.manylinux2010_x86_64.whl (1.6 MB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m1.6/1.6 MB\u001b[0m \u001b[31m115.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", "\u001b[?25hDownloading Markdown-3.6-py3-none-any.whl (105 kB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m105.4/105.4 kB\u001b[0m \u001b[31m20.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", "\u001b[?25hDownloading opencv_python_headless-4.10.0.82-cp37-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (49.9 MB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m49.9/49.9 MB\u001b[0m \u001b[31m93.7 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m:00:01\u001b[0m00:01\u001b[0m\n", "\u001b[?25hDownloading protobuf-4.25.3-cp37-abi3-manylinux2014_x86_64.whl (294 kB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m294.6/294.6 kB\u001b[0m \u001b[31m49.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", "\u001b[?25hDownloading pydantic-2.7.3-py3-none-any.whl (409 kB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m409.6/409.6 kB\u001b[0m \u001b[31m70.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", "\u001b[?25hDownloading pydantic_core-2.18.4-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (2.0 MB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m2.0/2.0 MB\u001b[0m \u001b[31m117.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", "\u001b[?25hDownloading pyparsing-3.1.2-py3-none-any.whl (103 kB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m103.2/103.2 kB\u001b[0m \u001b[31m21.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", "\u001b[?25hDownloading scikit_image-0.23.2-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (14.7 MB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m14.7/14.7 MB\u001b[0m \u001b[31m145.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0ma \u001b[36m0:00:01\u001b[0m\n", "\u001b[?25hDownloading scikit_learn-1.5.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (13.3 MB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m13.3/13.3 MB\u001b[0m \u001b[31m137.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m \u001b[36m0:00:01\u001b[0m\n", "\u001b[?25hDownloading tensorboard_data_server-0.7.2-py3-none-manylinux_2_31_x86_64.whl (6.6 MB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m6.6/6.6 MB\u001b[0m \u001b[31m96.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0mta \u001b[36m0:00:01\u001b[0m\n", "\u001b[?25hDownloading tzdata-2024.1-py2.py3-none-any.whl (345 kB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m345.4/345.4 kB\u001b[0m \u001b[31m51.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", "\u001b[?25hDownloading werkzeug-3.0.3-py3-none-any.whl (227 kB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m227.3/227.3 kB\u001b[0m \u001b[31m42.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", "\u001b[?25hDownloading annotated_types-0.7.0-py3-none-any.whl (13 kB)\n", "Downloading imageio-2.34.1-py3-none-any.whl (313 kB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m313.5/313.5 kB\u001b[0m \u001b[31m56.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", "\u001b[?25hDownloading joblib-1.4.2-py3-none-any.whl (301 kB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m301.8/301.8 kB\u001b[0m \u001b[31m49.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", "\u001b[?25hDownloading lazy_loader-0.4-py3-none-any.whl (12 kB)\n", "Downloading smmap-5.0.1-py3-none-any.whl (24 kB)\n", "Downloading threadpoolctl-3.5.0-py3-none-any.whl (18 kB)\n", "Downloading tifffile-2024.5.22-py3-none-any.whl (225 kB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m225.5/225.5 kB\u001b[0m \u001b[31m49.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", "\u001b[?25hInstalling collected packages: werkzeug, tzdata, tifffile, threadpoolctl, tensorboard-data-server, smmap, scipy, pyparsing, pydantic-core, protobuf, opencv-python-headless, opencv-python, markdown, lazy-loader, kiwisolver, joblib, imageio, grpcio, fonttools, cycler, contourpy, annotated-types, absl-py, tensorboard, scikit-learn, scikit-image, pydantic, pandas, matplotlib, gitdb, albucore, thop, seaborn, pycocotools, gitpython, albumentations\n", "Successfully installed absl-py-2.1.0 albucore-0.0.9 albumentations-1.4.8 annotated-types-0.7.0 contourpy-1.2.1 cycler-0.12.1 fonttools-4.53.0 gitdb-4.0.11 gitpython-3.1.43 grpcio-1.64.1 imageio-2.34.1 joblib-1.4.2 kiwisolver-1.4.5 lazy-loader-0.4 markdown-3.6 matplotlib-3.9.0 opencv-python-4.10.0.82 opencv-python-headless-4.10.0.82 pandas-2.2.2 protobuf-4.25.3 pycocotools-2.0.7 pydantic-2.7.3 pydantic-core-2.18.4 pyparsing-3.1.2 scikit-image-0.23.2 scikit-learn-1.5.0 scipy-1.13.1 seaborn-0.13.2 smmap-5.0.1 tensorboard-2.17.0 tensorboard-data-server-0.7.2 thop-0.1.1.post2209072238 threadpoolctl-3.5.0 tifffile-2024.5.22 tzdata-2024.1 werkzeug-3.0.3\n", "\u001b[33mWARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager. It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv\u001b[0m\u001b[33m\n", "\u001b[0m" ] } ], "source": [ "!pip install -r requirements.txt" ] }, { "cell_type": "code", "execution_count": 11, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Hit:1 https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64 InRelease\n", "Hit:2 http://archive.ubuntu.com/ubuntu jammy InRelease \n", "Hit:3 http://security.ubuntu.com/ubuntu jammy-security InRelease\n", "Hit:4 http://archive.ubuntu.com/ubuntu jammy-updates InRelease\n", "Hit:5 http://archive.ubuntu.com/ubuntu jammy-backports InRelease\n", "Reading package lists... Done\n", "Reading package lists... Done\n", "Building dependency tree... Done\n", "Reading state information... Done\n", "The following additional packages will be installed:\n", " libdrm-amdgpu1 libdrm-common libdrm-intel1 libdrm-nouveau2 libdrm-radeon1\n", " libdrm2 libgl1 libgl1-amber-dri libgl1-mesa-dri libglapi-mesa libglvnd0\n", " libglx-mesa0 libglx0 libllvm15 libpciaccess0 libsensors-config libsensors5\n", " libx11-6 libx11-data libx11-xcb1 libxau6 libxcb-dri2-0 libxcb-dri3-0\n", " libxcb-glx0 libxcb-present0 libxcb-randr0 libxcb-shm0 libxcb-sync1\n", " libxcb-xfixes0 libxcb1 libxdmcp6 libxext6 libxfixes3 libxshmfence1\n", " libxxf86vm1\n", "Suggested packages:\n", " pciutils lm-sensors\n", "The following NEW packages will be installed:\n", " libdrm-amdgpu1 libdrm-common libdrm-intel1 libdrm-nouveau2 libdrm-radeon1\n", " libdrm2 libgl1 libgl1-amber-dri libgl1-mesa-dri libgl1-mesa-glx\n", " libglapi-mesa libglvnd0 libglx-mesa0 libglx0 libllvm15 libpciaccess0\n", " libsensors-config libsensors5 libx11-6 libx11-data libx11-xcb1 libxau6\n", " libxcb-dri2-0 libxcb-dri3-0 libxcb-glx0 libxcb-present0 libxcb-randr0\n", " libxcb-shm0 libxcb-sync1 libxcb-xfixes0 libxcb1 libxdmcp6 libxext6\n", " libxfixes3 libxshmfence1 libxxf86vm1\n", "0 upgraded, 36 newly installed, 0 to remove and 56 not upgraded.\n", "Need to get 40.1 MB of archives.\n", "After this operation, 173 MB of additional disk space will be used.\n", "Get:1 http://archive.ubuntu.com/ubuntu jammy-updates/main amd64 libdrm-common all 2.4.113-2~ubuntu0.22.04.1 [5450 B]\n", "Get:2 http://archive.ubuntu.com/ubuntu jammy-updates/main amd64 libdrm2 amd64 2.4.113-2~ubuntu0.22.04.1 [38.1 kB]\n", "Get:3 http://archive.ubuntu.com/ubuntu jammy/main amd64 libxau6 amd64 1:1.0.9-1build5 [7634 B]\n", "Get:4 http://archive.ubuntu.com/ubuntu jammy/main amd64 libxdmcp6 amd64 1:1.1.3-0ubuntu5 [10.9 kB]\n", "Get:5 http://archive.ubuntu.com/ubuntu jammy/main amd64 libxcb1 amd64 1.14-3ubuntu3 [49.0 kB]\n", "Get:6 http://archive.ubuntu.com/ubuntu jammy-updates/main amd64 libx11-data all 2:1.7.5-1ubuntu0.3 [120 kB]\n", "Get:7 http://archive.ubuntu.com/ubuntu jammy-updates/main amd64 libx11-6 amd64 2:1.7.5-1ubuntu0.3 [667 kB]\n", "Get:8 http://archive.ubuntu.com/ubuntu jammy/main amd64 libxext6 amd64 2:1.3.4-1build1 [31.8 kB]\n", "Get:9 http://archive.ubuntu.com/ubuntu jammy-updates/main amd64 libdrm-amdgpu1 amd64 2.4.113-2~ubuntu0.22.04.1 [19.9 kB]\n", "Get:10 http://archive.ubuntu.com/ubuntu jammy/main amd64 libpciaccess0 amd64 0.16-3 [19.1 kB]\n", "Get:11 http://archive.ubuntu.com/ubuntu jammy-updates/main amd64 libdrm-intel1 amd64 2.4.113-2~ubuntu0.22.04.1 [66.7 kB]\n", "Get:12 http://archive.ubuntu.com/ubuntu jammy-updates/main amd64 libdrm-nouveau2 amd64 2.4.113-2~ubuntu0.22.04.1 [17.5 kB]\n", "Get:13 http://archive.ubuntu.com/ubuntu jammy-updates/main amd64 libdrm-radeon1 amd64 2.4.113-2~ubuntu0.22.04.1 [21.6 kB]\n", "Get:14 http://archive.ubuntu.com/ubuntu jammy-updates/main amd64 libglapi-mesa amd64 23.2.1-1ubuntu3.1~22.04.2 [37.1 kB]\n", "Get:15 http://archive.ubuntu.com/ubuntu jammy-updates/main amd64 libgl1-amber-dri amd64 21.3.9-0ubuntu1~22.04.1 [4218 kB]\n", "Get:16 http://archive.ubuntu.com/ubuntu jammy-updates/main amd64 libllvm15 amd64 1:15.0.7-0ubuntu0.22.04.3 [25.4 MB]\n", "Get:17 http://archive.ubuntu.com/ubuntu jammy/main amd64 libsensors-config all 1:3.6.0-7ubuntu1 [5274 B]\n", "Get:18 http://archive.ubuntu.com/ubuntu jammy/main amd64 libsensors5 amd64 1:3.6.0-7ubuntu1 [26.3 kB]\n", "Get:19 http://archive.ubuntu.com/ubuntu jammy/main amd64 libxcb-dri3-0 amd64 1.14-3ubuntu3 [6968 B]\n", "Get:20 http://archive.ubuntu.com/ubuntu jammy-updates/main amd64 libgl1-mesa-dri amd64 23.2.1-1ubuntu3.1~22.04.2 [8860 kB]\n", "Get:21 http://archive.ubuntu.com/ubuntu jammy/main amd64 libglvnd0 amd64 1.4.0-1 [73.6 kB]\n", "Get:22 http://archive.ubuntu.com/ubuntu jammy-updates/main amd64 libx11-xcb1 amd64 2:1.7.5-1ubuntu0.3 [7802 B]\n", "Get:23 http://archive.ubuntu.com/ubuntu jammy/main amd64 libxcb-dri2-0 amd64 1.14-3ubuntu3 [7206 B]\n", "Get:24 http://archive.ubuntu.com/ubuntu jammy/main amd64 libxcb-glx0 amd64 1.14-3ubuntu3 [25.9 kB]\n", "Get:25 http://archive.ubuntu.com/ubuntu jammy/main amd64 libxcb-present0 amd64 1.14-3ubuntu3 [5734 B]\n", "Get:26 http://archive.ubuntu.com/ubuntu jammy/main amd64 libxcb-randr0 amd64 1.14-3ubuntu3 [18.3 kB]\n", "Get:27 http://archive.ubuntu.com/ubuntu jammy/main amd64 libxcb-shm0 amd64 1.14-3ubuntu3 [5780 B]\n", "Get:28 http://archive.ubuntu.com/ubuntu jammy/main amd64 libxcb-sync1 amd64 1.14-3ubuntu3 [9416 B]\n", "Get:29 http://archive.ubuntu.com/ubuntu jammy/main amd64 libxcb-xfixes0 amd64 1.14-3ubuntu3 [9996 B]\n", "Get:30 http://archive.ubuntu.com/ubuntu jammy/main amd64 libxfixes3 amd64 1:6.0.0-1 [11.7 kB]\n", "Get:31 http://archive.ubuntu.com/ubuntu jammy/main amd64 libxshmfence1 amd64 1.3-1build4 [5394 B]\n", "Get:32 http://archive.ubuntu.com/ubuntu jammy/main amd64 libxxf86vm1 amd64 1:1.1.4-1build3 [10.4 kB]\n", "Get:33 http://archive.ubuntu.com/ubuntu jammy-updates/main amd64 libglx-mesa0 amd64 23.2.1-1ubuntu3.1~22.04.2 [158 kB]\n", "Get:34 http://archive.ubuntu.com/ubuntu jammy/main amd64 libglx0 amd64 1.4.0-1 [41.0 kB]\n", "Get:35 http://archive.ubuntu.com/ubuntu jammy/main amd64 libgl1 amd64 1.4.0-1 [110 kB]\n", "Get:36 http://archive.ubuntu.com/ubuntu jammy-updates/universe amd64 libgl1-mesa-glx amd64 23.0.4-0ubuntu1~22.04.1 [5584 B]\n", "Fetched 40.1 MB in 3s (14.7 MB/s)\n", "debconf: delaying package configuration, since apt-utils is not installed\n", "Selecting previously unselected package libdrm-common.\n", "(Reading database ... 22469 files and directories currently installed.)\n", "Preparing to unpack .../00-libdrm-common_2.4.113-2~ubuntu0.22.04.1_all.deb ...\n", "Unpacking libdrm-common (2.4.113-2~ubuntu0.22.04.1) ...\n", "Selecting previously unselected package libdrm2:amd64.\n", "Preparing to unpack .../01-libdrm2_2.4.113-2~ubuntu0.22.04.1_amd64.deb ...\n", "Unpacking libdrm2:amd64 (2.4.113-2~ubuntu0.22.04.1) ...\n", "Selecting previously unselected package libxau6:amd64.\n", "Preparing to unpack .../02-libxau6_1%3a1.0.9-1build5_amd64.deb ...\n", "Unpacking libxau6:amd64 (1:1.0.9-1build5) ...\n", "Selecting previously unselected package libxdmcp6:amd64.\n", "Preparing to unpack .../03-libxdmcp6_1%3a1.1.3-0ubuntu5_amd64.deb ...\n", "Unpacking libxdmcp6:amd64 (1:1.1.3-0ubuntu5) ...\n", "Selecting previously unselected package libxcb1:amd64.\n", "Preparing to unpack .../04-libxcb1_1.14-3ubuntu3_amd64.deb ...\n", "Unpacking libxcb1:amd64 (1.14-3ubuntu3) ...\n", "Selecting previously unselected package libx11-data.\n", "Preparing to unpack .../05-libx11-data_2%3a1.7.5-1ubuntu0.3_all.deb ...\n", "Unpacking libx11-data (2:1.7.5-1ubuntu0.3) ...\n", "Selecting previously unselected package libx11-6:amd64.\n", "Preparing to unpack .../06-libx11-6_2%3a1.7.5-1ubuntu0.3_amd64.deb ...\n", "Unpacking libx11-6:amd64 (2:1.7.5-1ubuntu0.3) ...\n", "Selecting previously unselected package libxext6:amd64.\n", "Preparing to unpack .../07-libxext6_2%3a1.3.4-1build1_amd64.deb ...\n", "Unpacking libxext6:amd64 (2:1.3.4-1build1) ...\n", "Selecting previously unselected package libdrm-amdgpu1:amd64.\n", "Preparing to unpack .../08-libdrm-amdgpu1_2.4.113-2~ubuntu0.22.04.1_amd64.deb ...\n", "Unpacking libdrm-amdgpu1:amd64 (2.4.113-2~ubuntu0.22.04.1) ...\n", "Selecting previously unselected package libpciaccess0:amd64.\n", "Preparing to unpack .../09-libpciaccess0_0.16-3_amd64.deb ...\n", "Unpacking libpciaccess0:amd64 (0.16-3) ...\n", "Selecting previously unselected package libdrm-intel1:amd64.\n", "Preparing to unpack .../10-libdrm-intel1_2.4.113-2~ubuntu0.22.04.1_amd64.deb ...\n", "Unpacking libdrm-intel1:amd64 (2.4.113-2~ubuntu0.22.04.1) ...\n", "Selecting previously unselected package libdrm-nouveau2:amd64.\n", "Preparing to unpack .../11-libdrm-nouveau2_2.4.113-2~ubuntu0.22.04.1_amd64.deb ...\n", "Unpacking libdrm-nouveau2:amd64 (2.4.113-2~ubuntu0.22.04.1) ...\n", "Selecting previously unselected package libdrm-radeon1:amd64.\n", "Preparing to unpack .../12-libdrm-radeon1_2.4.113-2~ubuntu0.22.04.1_amd64.deb ...\n", "Unpacking libdrm-radeon1:amd64 (2.4.113-2~ubuntu0.22.04.1) ...\n", "Selecting previously unselected package libglapi-mesa:amd64.\n", "Preparing to unpack .../13-libglapi-mesa_23.2.1-1ubuntu3.1~22.04.2_amd64.deb ...\n", "Unpacking libglapi-mesa:amd64 (23.2.1-1ubuntu3.1~22.04.2) ...\n", "Selecting previously unselected package libgl1-amber-dri:amd64.\n", "Preparing to unpack .../14-libgl1-amber-dri_21.3.9-0ubuntu1~22.04.1_amd64.deb ...\n", "Unpacking libgl1-amber-dri:amd64 (21.3.9-0ubuntu1~22.04.1) ...\n", "Selecting previously unselected package libllvm15:amd64.\n", "Preparing to unpack .../15-libllvm15_1%3a15.0.7-0ubuntu0.22.04.3_amd64.deb ...\n", "Unpacking libllvm15:amd64 (1:15.0.7-0ubuntu0.22.04.3) ...\n", "Selecting previously unselected package libsensors-config.\n", "Preparing to unpack .../16-libsensors-config_1%3a3.6.0-7ubuntu1_all.deb ...\n", "Unpacking libsensors-config (1:3.6.0-7ubuntu1) ...\n", "Selecting previously unselected package libsensors5:amd64.\n", "Preparing to unpack .../17-libsensors5_1%3a3.6.0-7ubuntu1_amd64.deb ...\n", "Unpacking libsensors5:amd64 (1:3.6.0-7ubuntu1) ...\n", "Selecting previously unselected package libxcb-dri3-0:amd64.\n", "Preparing to unpack .../18-libxcb-dri3-0_1.14-3ubuntu3_amd64.deb ...\n", "Unpacking libxcb-dri3-0:amd64 (1.14-3ubuntu3) ...\n", "Selecting previously unselected package libgl1-mesa-dri:amd64.\n", "Preparing to unpack .../19-libgl1-mesa-dri_23.2.1-1ubuntu3.1~22.04.2_amd64.deb ...\n", "Unpacking libgl1-mesa-dri:amd64 (23.2.1-1ubuntu3.1~22.04.2) ...\n", "Selecting previously unselected package libglvnd0:amd64.\n", "Preparing to unpack .../20-libglvnd0_1.4.0-1_amd64.deb ...\n", "Unpacking libglvnd0:amd64 (1.4.0-1) ...\n", "Selecting previously unselected package libx11-xcb1:amd64.\n", "Preparing to unpack .../21-libx11-xcb1_2%3a1.7.5-1ubuntu0.3_amd64.deb ...\n", "Unpacking libx11-xcb1:amd64 (2:1.7.5-1ubuntu0.3) ...\n", "Selecting previously unselected package libxcb-dri2-0:amd64.\n", "Preparing to unpack .../22-libxcb-dri2-0_1.14-3ubuntu3_amd64.deb ...\n", "Unpacking libxcb-dri2-0:amd64 (1.14-3ubuntu3) ...\n", "Selecting previously unselected package libxcb-glx0:amd64.\n", "Preparing to unpack .../23-libxcb-glx0_1.14-3ubuntu3_amd64.deb ...\n", "Unpacking libxcb-glx0:amd64 (1.14-3ubuntu3) ...\n", "Selecting previously unselected package libxcb-present0:amd64.\n", "Preparing to unpack .../24-libxcb-present0_1.14-3ubuntu3_amd64.deb ...\n", "Unpacking libxcb-present0:amd64 (1.14-3ubuntu3) ...\n", "Selecting previously unselected package libxcb-randr0:amd64.\n", "Preparing to unpack .../25-libxcb-randr0_1.14-3ubuntu3_amd64.deb ...\n", "Unpacking libxcb-randr0:amd64 (1.14-3ubuntu3) ...\n", "Selecting previously unselected package libxcb-shm0:amd64.\n", "Preparing to unpack .../26-libxcb-shm0_1.14-3ubuntu3_amd64.deb ...\n", "Unpacking libxcb-shm0:amd64 (1.14-3ubuntu3) ...\n", "Selecting previously unselected package libxcb-sync1:amd64.\n", "Preparing to unpack .../27-libxcb-sync1_1.14-3ubuntu3_amd64.deb ...\n", "Unpacking libxcb-sync1:amd64 (1.14-3ubuntu3) ...\n", "Selecting previously unselected package libxcb-xfixes0:amd64.\n", "Preparing to unpack .../28-libxcb-xfixes0_1.14-3ubuntu3_amd64.deb ...\n", "Unpacking libxcb-xfixes0:amd64 (1.14-3ubuntu3) ...\n", "Selecting previously unselected package libxfixes3:amd64.\n", "Preparing to unpack .../29-libxfixes3_1%3a6.0.0-1_amd64.deb ...\n", "Unpacking libxfixes3:amd64 (1:6.0.0-1) ...\n", "Selecting previously unselected package libxshmfence1:amd64.\n", "Preparing to unpack .../30-libxshmfence1_1.3-1build4_amd64.deb ...\n", "Unpacking libxshmfence1:amd64 (1.3-1build4) ...\n", "Selecting previously unselected package libxxf86vm1:amd64.\n", "Preparing to unpack .../31-libxxf86vm1_1%3a1.1.4-1build3_amd64.deb ...\n", "Unpacking libxxf86vm1:amd64 (1:1.1.4-1build3) ...\n", "Selecting previously unselected package libglx-mesa0:amd64.\n", "Preparing to unpack .../32-libglx-mesa0_23.2.1-1ubuntu3.1~22.04.2_amd64.deb ...\n", "Unpacking libglx-mesa0:amd64 (23.2.1-1ubuntu3.1~22.04.2) ...\n", "Selecting previously unselected package libglx0:amd64.\n", "Preparing to unpack .../33-libglx0_1.4.0-1_amd64.deb ...\n", "Unpacking libglx0:amd64 (1.4.0-1) ...\n", "Selecting previously unselected package libgl1:amd64.\n", "Preparing to unpack .../34-libgl1_1.4.0-1_amd64.deb ...\n", "Unpacking libgl1:amd64 (1.4.0-1) ...\n", "Selecting previously unselected package libgl1-mesa-glx:amd64.\n", "Preparing to unpack .../35-libgl1-mesa-glx_23.0.4-0ubuntu1~22.04.1_amd64.deb ...\n", "Unpacking libgl1-mesa-glx:amd64 (23.0.4-0ubuntu1~22.04.1) ...\n", "Setting up libpciaccess0:amd64 (0.16-3) ...\n", "Setting up libxau6:amd64 (1:1.0.9-1build5) ...\n", "Setting up libxdmcp6:amd64 (1:1.1.3-0ubuntu5) ...\n", "Setting up libxcb1:amd64 (1.14-3ubuntu3) ...\n", "Setting up libxcb-xfixes0:amd64 (1.14-3ubuntu3) ...\n", "Setting up libglvnd0:amd64 (1.4.0-1) ...\n", "Setting up libxcb-glx0:amd64 (1.14-3ubuntu3) ...\n", "Setting up libsensors-config (1:3.6.0-7ubuntu1) ...\n", "Setting up libxcb-shm0:amd64 (1.14-3ubuntu3) ...\n", "Setting up libxcb-present0:amd64 (1.14-3ubuntu3) ...\n", "Setting up libx11-data (2:1.7.5-1ubuntu0.3) ...\n", "Setting up libxcb-sync1:amd64 (1.14-3ubuntu3) ...\n", "Setting up libsensors5:amd64 (1:3.6.0-7ubuntu1) ...\n", "Setting up libglapi-mesa:amd64 (23.2.1-1ubuntu3.1~22.04.2) ...\n", "Setting up libxcb-dri2-0:amd64 (1.14-3ubuntu3) ...\n", "Setting up libxshmfence1:amd64 (1.3-1build4) ...\n", "Setting up libxcb-randr0:amd64 (1.14-3ubuntu3) ...\n", "Setting up libllvm15:amd64 (1:15.0.7-0ubuntu0.22.04.3) ...\n", "Setting up libx11-6:amd64 (2:1.7.5-1ubuntu0.3) ...\n", "Setting up libdrm-common (2.4.113-2~ubuntu0.22.04.1) ...\n", "Setting up libxcb-dri3-0:amd64 (1.14-3ubuntu3) ...\n", "Setting up libx11-xcb1:amd64 (2:1.7.5-1ubuntu0.3) ...\n", "Setting up libxext6:amd64 (2:1.3.4-1build1) ...\n", "Setting up libxxf86vm1:amd64 (1:1.1.4-1build3) ...\n", "Setting up libxfixes3:amd64 (1:6.0.0-1) ...\n", "Setting up libdrm2:amd64 (2.4.113-2~ubuntu0.22.04.1) ...\n", "Setting up libdrm-amdgpu1:amd64 (2.4.113-2~ubuntu0.22.04.1) ...\n", "Setting up libdrm-nouveau2:amd64 (2.4.113-2~ubuntu0.22.04.1) ...\n", "Setting up libdrm-radeon1:amd64 (2.4.113-2~ubuntu0.22.04.1) ...\n", "Setting up libdrm-intel1:amd64 (2.4.113-2~ubuntu0.22.04.1) ...\n", "Setting up libgl1-mesa-dri:amd64 (23.2.1-1ubuntu3.1~22.04.2) ...\n", "Setting up libgl1-amber-dri:amd64 (21.3.9-0ubuntu1~22.04.1) ...\n", "Setting up libglx-mesa0:amd64 (23.2.1-1ubuntu3.1~22.04.2) ...\n", "Setting up libglx0:amd64 (1.4.0-1) ...\n", "Setting up libgl1:amd64 (1.4.0-1) ...\n", "Setting up libgl1-mesa-glx:amd64 (23.0.4-0ubuntu1~22.04.1) ...\n", "Processing triggers for libc-bin (2.35-0ubuntu3.4) ...\n", "yes: standard output: Broken pipe\n" ] } ], "source": [ "!sudo apt-get update\n", "!yes | sudo apt-get install libgl1-mesa-glx" ] }, { "cell_type": "code", "execution_count": 9, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "/workspace/yolov9\n" ] } ], "source": [ "%cd /workspace/yolov9" ] }, { "cell_type": "code", "execution_count": 22, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Collecting Pillow==9.5.0\n", " Downloading Pillow-9.5.0-cp310-cp310-manylinux_2_28_x86_64.whl.metadata (9.5 kB)\n", "Downloading Pillow-9.5.0-cp310-cp310-manylinux_2_28_x86_64.whl (3.4 MB)\n", "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m3.4/3.4 MB\u001b[0m \u001b[31m54.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0ma \u001b[36m0:00:01\u001b[0m\n", "\u001b[?25hInstalling collected packages: Pillow\n", " Attempting uninstall: Pillow\n", " Found existing installation: Pillow 10.0.1\n", " Uninstalling Pillow-10.0.1:\n", " Successfully uninstalled Pillow-10.0.1\n", "Successfully installed Pillow-9.5.0\n", "\u001b[33mWARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager. It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv\u001b[0m\u001b[33m\n", "\u001b[0mNote: you may need to restart the kernel to use updated packages.\n" ] } ], "source": [ "pip install Pillow==9.5.0" ] }, { "cell_type": "code", "execution_count": 23, "metadata": { "colab": { "base_uri": "https://localhost:8080/" }, "id": "jjB9EA9yRCyR", "outputId": "2e388fee-6ec5-492d-b698-55d2d1cec2f2" }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "\u001b[34m\u001b[1mtrain_dual: \u001b[0mweights=, cfg=/workspace/YOLO9_KMUTT/yolov9/models/detect/yolov9-kmuttX.yaml, data=/workspace/6400 images/data.yaml, hyp=hyp.scratch-high.yaml, epochs=500, batch_size=4, imgsz=640, rect=False, resume=False, nosave=False, noval=False, noautoanchor=False, noplots=False, evolve=None, bucket=, cache=None, image_weights=False, device=0, multi_scale=False, single_cls=False, optimizer=SGD, sync_bn=False, workers=8, project=runs/train, name=yolov9-c, exist_ok=False, quad=False, cos_lr=False, flat_cos_lr=False, fixed_lr=False, label_smoothing=0.0, patience=100, freeze=[0], save_period=-1, seed=0, local_rank=-1, min_items=0, close_mosaic=15, entity=None, upload_dataset=False, bbox_interval=-1, artifact_alias=latest\n", "YOLO 🚀 2024-6-7 Python-3.10.13 torch-2.2.0 CUDA:0 (NVIDIA GeForce RTX 4090, 24210MiB)\n", "\n", "\u001b[34m\u001b[1mhyperparameters: \u001b[0mlr0=0.01, lrf=0.01, momentum=0.937, weight_decay=0.0005, warmup_epochs=3.0, warmup_momentum=0.8, warmup_bias_lr=0.1, box=7.5, cls=0.5, cls_pw=1.0, obj=0.7, obj_pw=1.0, dfl=1.5, iou_t=0.2, anchor_t=5.0, fl_gamma=0.0, hsv_h=0.015, hsv_s=0.7, hsv_v=0.4, degrees=0.0, translate=0.1, scale=0.9, shear=0.0, perspective=0.0, flipud=0.0, fliplr=0.5, mosaic=1.0, mixup=0.15, copy_paste=0.3\n", "\u001b[34m\u001b[1mClearML: \u001b[0mrun 'pip install clearml' to automatically track, visualize and remotely train YOLO 🚀 in ClearML\n", "\u001b[34m\u001b[1mComet: \u001b[0mrun 'pip install comet_ml' to automatically track and visualize YOLO 🚀 runs in Comet\n", "\u001b[34m\u001b[1mTensorBoard: \u001b[0mStart with 'tensorboard --logdir runs/train', view at http://localhost:6006/\n", "Overriding model.yaml nc=80 with nc=10\n", "\n", " from n params module arguments \n", " 0 -1 1 0 models.common.Silence [] \n", " 1 -1 1 1856 models.common.Conv [3, 64, 3, 2] \n", " 2 -1 1 73984 models.common.Conv [64, 128, 3, 2] \n", " 3 -1 1 252160 models.common.RepNCSPELAN4 [128, 256, 128, 64, 2] \n", " 4 -1 1 164352 models.common.ADown [256, 256] \n", " 5 -1 1 1004032 models.common.RepNCSPELAN4 [256, 512, 256, 128, 2] \n", " 6 -1 1 656384 models.common.ADown [512, 512] \n", " 7 -1 1 4006912 models.common.RepNCSPELAN4 [512, 1024, 512, 256, 2] \n", " 8 -1 1 2623488 models.common.ADown [1024, 1024] \n", " 9 -1 1 4269056 models.common.RepNCSPELAN4 [1024, 1024, 512, 256, 2] \n", " 10 1 1 4160 models.common.CBLinear [64, [64]] \n", " 11 3 1 49344 models.common.CBLinear [256, [64, 128]] \n", " 12 5 1 229824 models.common.CBLinear [512, [64, 128, 256]] \n", " 13 7 1 984000 models.common.CBLinear [1024, [64, 128, 256, 512]] \n", " 14 9 1 2033600 models.common.CBLinear [1024, [64, 128, 256, 512, 1024]]\n", " 15 0 1 1856 models.common.Conv [3, 64, 3, 2] \n", " 16[10, 11, 12, 13, 14, -1] 1 0 models.common.CBFuse [[0, 0, 0, 0, 0]] \n", " 17 -1 1 73984 models.common.Conv [64, 128, 3, 2] \n", " 18[11, 12, 13, 14, -1] 1 0 models.common.CBFuse [[1, 1, 1, 1]] \n", " 19 -1 1 252160 models.common.RepNCSPELAN4 [128, 256, 128, 64, 2] \n", " 20 -1 1 164352 models.common.ADown [256, 256] \n", " 21 [12, 13, 14, -1] 1 0 models.common.CBFuse [[2, 2, 2]] \n", " 22 -1 1 1004032 models.common.RepNCSPELAN4 [256, 512, 256, 128, 2] \n", " 23 -1 1 656384 models.common.ADown [512, 512] \n", " 24 [13, 14, -1] 1 0 models.common.CBFuse [[3, 3]] \n", " 25 -1 1 4006912 models.common.RepNCSPELAN4 [512, 1024, 512, 256, 2] \n", " 26 -1 1 2623488 models.common.ADown [1024, 1024] \n", " 27 [14, -1] 1 0 models.common.CBFuse [[4]] \n", " 28 -1 1 4269056 models.common.RepNCSPELAN4 [1024, 1024, 512, 256, 2] \n", " 29 9 1 787968 models.common.SPPELAN [1024, 512, 256] \n", " 30 -1 1 0 torch.nn.modules.upsampling.Upsample [None, 2, 'nearest'] \n", " 31 [-1, 7] 1 0 models.common.Concat [1] \n", " 32 -1 1 4005888 models.common.RepNCSPELAN4 [1536, 512, 512, 256, 2] \n", " 33 -1 1 0 torch.nn.modules.upsampling.Upsample [None, 2, 'nearest'] \n", " 34 [-1, 5] 1 0 models.common.Concat [1] \n", " 35 -1 1 1069056 models.common.RepNCSPELAN4 [1024, 256, 256, 128, 2] \n", " 36 28 1 787968 models.common.SPPELAN [1024, 512, 256] \n", " 37 -1 1 0 torch.nn.modules.upsampling.Upsample [None, 2, 'nearest'] \n", " 38 [-1, 25] 1 0 models.common.Concat [1] \n", " 39 -1 1 4005888 models.common.RepNCSPELAN4 [1536, 512, 512, 256, 2] \n", " 40 -1 1 0 torch.nn.modules.upsampling.Upsample [None, 2, 'nearest'] \n", " 41 [-1, 22] 1 0 models.common.Concat [1] \n", " 42 -1 1 1069056 models.common.RepNCSPELAN4 [1024, 256, 256, 128, 2] \n", " 43 -1 1 164352 models.common.ADown [256, 256] \n", " 44 [-1, 39] 1 0 models.common.Concat [1] \n", " 45 -1 1 3612672 models.common.RepNCSPELAN4 [768, 512, 512, 256, 2] \n", " 46 -1 1 656384 models.common.ADown [512, 512] \n", " 47 [-1, 36] 1 0 models.common.Concat [1] \n", " 48 -1 1 13910016 models.common.RepNCSPELAN4 [1024, 1024, 1024, 512, 2] \n", " 49 -1 1 3948032 models.common.PSA [1024, 1024] \n", " 50 35 1 249728 models.common.PSA [256, 256] \n", " 51 32 1 990976 models.common.PSA [512, 512] \n", " 52 29 1 990976 models.common.PSA [512, 512] \n", " 53 42 1 249728 models.common.PSA [256, 256] \n", " 54 45 1 990976 models.common.PSA [512, 512] \n", " 55[50, 51, 52, 53, 54, 49] 1 12471260 models.yolo.DualDDetect [10, [256, 512, 512, 256, 512, 1024]]\n", "yolov9-kmuttX summary: 1643 layers, 79366300 parameters, 79366268 gradients, 263.8 GFLOPs\n", "\n", "\u001b[34m\u001b[1mAMP: \u001b[0mchecks passed ✅\n", "\u001b[34m\u001b[1moptimizer:\u001b[0m SGD(lr=0.01) with parameter groups 398 weight(decay=0.0), 417 weight(decay=0.0005), 415 bias\n", "\u001b[34m\u001b[1malbumentations: \u001b[0mBlur(p=0.01, blur_limit=(3, 7)), MedianBlur(p=0.01, blur_limit=(3, 7)), ToGray(p=0.01), CLAHE(p=0.01, clip_limit=(1, 4.0), tile_grid_size=(8, 8))\n", "\u001b[34m\u001b[1mtrain: \u001b[0mScanning /workspace/6400 images/train/labels.cache... 5120 images, 7 back\u001b[0m\n", "\u001b[34m\u001b[1mval: \u001b[0mScanning /workspace/6400 images/valid/labels.cache... 640 images, 0 backgro\u001b[0m\n", "Plotting labels to runs/train/yolov9-c10/labels.jpg... \n", "Image sizes 640 train, 640 val\n", "Using 4 dataloader workers\n", "Logging results to \u001b[1mruns/train/yolov9-c10\u001b[0m\n", "Starting training for 500 epochs...\n", "\n", " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", " 0/499 16.5G 4.939 7.156 5.232 11 640: WARNING ⚠️ TensorBoard graph visualization failure Only tensors, lists, tuples of tensors, or dictionary of tensors can be output from traced functions\n", " 0/499 16.7G 5.236 7.764 5.234 23 640: ^C\n", " 0/499 16.7G 5.236 7.764 5.234 23 640: \n", "Traceback (most recent call last):\n", " File \"/workspace/YOLO9_KMUTT/yolov9/train_dual.py\", line 644, in \n", " main(opt)\n", " File \"/workspace/YOLO9_KMUTT/yolov9/train_dual.py\", line 538, in main\n", " train(opt.hyp, opt, device, callbacks)\n", " File \"/workspace/YOLO9_KMUTT/yolov9/train_dual.py\", line 314, in train\n", " pred = model(imgs) # forward\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1511, in _wrapped_call_impl\n", " return self._call_impl(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1520, in _call_impl\n", " return forward_call(*args, **kwargs)\n", " File \"/workspace/YOLO9_KMUTT/yolov9/models/yolo.py\", line 633, in forward\n", " return self._forward_once(x, profile, visualize) # single-scale inference, train\n", " File \"/workspace/YOLO9_KMUTT/yolov9/models/yolo.py\", line 533, in _forward_once\n", " x = m(x) # run\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1511, in _wrapped_call_impl\n", " return self._call_impl(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1520, in _call_impl\n", " return forward_call(*args, **kwargs)\n", " File \"/workspace/YOLO9_KMUTT/yolov9/models/common.py\", line 613, in forward\n", " y.extend((m(y[-1])) for m in [self.cv2, self.cv3])\n", " File \"/workspace/YOLO9_KMUTT/yolov9/models/common.py\", line 613, in \n", " y.extend((m(y[-1])) for m in [self.cv2, self.cv3])\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1511, in _wrapped_call_impl\n", " return self._call_impl(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1520, in _call_impl\n", " return forward_call(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/container.py\", line 217, in forward\n", " input = module(input)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1511, in _wrapped_call_impl\n", " return self._call_impl(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1520, in _call_impl\n", " return forward_call(*args, **kwargs)\n", " File \"/workspace/YOLO9_KMUTT/yolov9/models/common.py\", line 384, in forward\n", " return self.cv3(torch.cat((self.m(self.cv1(x)), self.cv2(x)), 1))\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1511, in _wrapped_call_impl\n", " return self._call_impl(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1520, in _call_impl\n", " return forward_call(*args, **kwargs)\n", " File \"/workspace/YOLO9_KMUTT/yolov9/models/common.py\", line 54, in forward\n", " return self.act(self.bn(self.conv(x)))\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1511, in _wrapped_call_impl\n", " return self._call_impl(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1520, in _call_impl\n", " return forward_call(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/batchnorm.py\", line 175, in forward\n", " return F.batch_norm(\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/functional.py\", line 2482, in batch_norm\n", " return torch.batch_norm(\n", "KeyboardInterrupt\n" ] } ], "source": [ "!python train_dual.py --workers 8 --device 0 --batch 4 --data '/workspace/6400 images/data.yaml' --img 640 --cfg /workspace/YOLO9_KMUTT/yolov9/models/detect/yolov9-kmuttX.yaml --weights '' --name yolov9-c --hyp hyp.scratch-high.yaml --min-items 0 --epochs 500 --close-mosaic 15" ] }, { "cell_type": "code", "execution_count": 9, "metadata": { "id": "VRMBngDDsIok" }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "[2024-06-01 23:55:21,192] torch.distributed.run: [WARNING] \n", "[2024-06-01 23:55:21,192] torch.distributed.run: [WARNING] *****************************************\n", "[2024-06-01 23:55:21,192] torch.distributed.run: [WARNING] Setting OMP_NUM_THREADS environment variable for each process to be 1 in default, to avoid your system being overloaded, please further tune the variable for optimal performance in your application as needed. \n", "[2024-06-01 23:55:21,192] torch.distributed.run: [WARNING] *****************************************\n", "\u001b[34m\u001b[1mtrain_dual: \u001b[0mweights=, cfg=models/detect/yolov9-c.yaml, data=/workspace/norway/data.yaml, hyp=hyp.scratch-high.yaml, epochs=500, batch_size=30, imgsz=640, rect=False, resume=False, nosave=False, noval=False, noautoanchor=False, noplots=False, evolve=None, bucket=, cache=None, image_weights=False, device=0,1, multi_scale=False, single_cls=False, optimizer=SGD, sync_bn=True, workers=8, project=runs/train, name=yolov9-c, exist_ok=False, quad=False, cos_lr=False, flat_cos_lr=False, fixed_lr=False, label_smoothing=0.0, patience=100, freeze=[0], save_period=-1, seed=0, local_rank=-1, min_items=0, close_mosaic=15, entity=None, upload_dataset=False, bbox_interval=-1, artifact_alias=latest\n", "Traceback (most recent call last):\n", " File \"/workspace/yolov9/train_dual.py\", line 644, in \n", " main(opt)\n", " File \"/workspace/yolov9/train_dual.py\", line 530, in main\n", " assert opt.batch_size % WORLD_SIZE == 0, f'--batch-size {opt.batch_size} must be multiple of WORLD_SIZE'\n", "AssertionError: --batch-size 30 must be multiple of WORLD_SIZE\n", "Traceback (most recent call last):\n", " File \"/workspace/yolov9/train_dual.py\", line 644, in \n", " main(opt)\n", " File \"/workspace/yolov9/train_dual.py\", line 530, in main\n", " assert opt.batch_size % WORLD_SIZE == 0, f'--batch-size {opt.batch_size} must be multiple of WORLD_SIZE'\n", "AssertionError: --batch-size 30 must be multiple of WORLD_SIZE\n", "Traceback (most recent call last):\n", " File \"/workspace/yolov9/train_dual.py\", line 644, in \n", " main(opt)\n", " File \"/workspace/yolov9/train_dual.py\", line 530, in main\n", " assert opt.batch_size % WORLD_SIZE == 0, f'--batch-size {opt.batch_size} must be multiple of WORLD_SIZE'\n", "AssertionError: --batch-size 30 must be multiple of WORLD_SIZE\n", "YOLO 🚀 v0.1-94-gcd88016 Python-3.10.13 torch-2.2.0 CUDA:0 (NVIDIA GeForce RTX 4090, 24217MiB)\n", " CUDA:1 (NVIDIA GeForce RTX 4090, 24217MiB)\n", "\n", "Traceback (most recent call last):\n", " File \"/workspace/yolov9/train_dual.py\", line 644, in \n", " main(opt)\n", " File \"/workspace/yolov9/train_dual.py\", line 530, in main\n", " assert opt.batch_size % WORLD_SIZE == 0, f'--batch-size {opt.batch_size} must be multiple of WORLD_SIZE'\n", "AssertionError: --batch-size 30 must be multiple of WORLD_SIZE\n", "Traceback (most recent call last):\n", " File \"/workspace/yolov9/train_dual.py\", line 644, in \n", " main(opt)\n", " File \"/workspace/yolov9/train_dual.py\", line 530, in main\n", " assert opt.batch_size % WORLD_SIZE == 0, f'--batch-size {opt.batch_size} must be multiple of WORLD_SIZE'\n", "AssertionError: --batch-size 30 must be multiple of WORLD_SIZE\n", "Traceback (most recent call last):\n", " File \"/workspace/yolov9/train_dual.py\", line 644, in \n", " main(opt)\n", " File \"/workspace/yolov9/train_dual.py\", line 530, in main\n", " assert opt.batch_size % WORLD_SIZE == 0, f'--batch-size {opt.batch_size} must be multiple of WORLD_SIZE'\n", "AssertionError: --batch-size 30 must be multiple of WORLD_SIZE\n", "Traceback (most recent call last):\n", " File \"/workspace/yolov9/train_dual.py\", line 644, in \n", " main(opt)\n", " File \"/workspace/yolov9/train_dual.py\", line 530, in main\n", " assert opt.batch_size % WORLD_SIZE == 0, f'--batch-size {opt.batch_size} must be multiple of WORLD_SIZE'\n", "AssertionError: --batch-size 30 must be multiple of WORLD_SIZE\n", "Traceback (most recent call last):\n", " File \"/workspace/yolov9/train_dual.py\", line 644, in \n", " main(opt)\n", " File \"/workspace/yolov9/train_dual.py\", line 530, in main\n", " assert opt.batch_size % WORLD_SIZE == 0, f'--batch-size {opt.batch_size} must be multiple of WORLD_SIZE'\n", "AssertionError: --batch-size 30 must be multiple of WORLD_SIZE\n", "[2024-06-01 23:55:26,209] torch.distributed.elastic.multiprocessing.api: [ERROR] failed (exitcode: 1) local_rank: 0 (pid: 4360) of binary: /opt/conda/bin/python\n", "Traceback (most recent call last):\n", " File \"/opt/conda/bin/torchrun\", line 33, in \n", " sys.exit(load_entry_point('torch==2.2.0', 'console_scripts', 'torchrun')())\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/distributed/elastic/multiprocessing/errors/__init__.py\", line 347, in wrapper\n", " return f(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/distributed/run.py\", line 812, in main\n", " run(args)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/distributed/run.py\", line 803, in run\n", " elastic_launch(\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/distributed/launcher/api.py\", line 135, in __call__\n", " return launch_agent(self._config, self._entrypoint, list(args))\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/distributed/launcher/api.py\", line 268, in launch_agent\n", " raise ChildFailedError(\n", "torch.distributed.elastic.multiprocessing.errors.ChildFailedError: \n", "============================================================\n", "train_dual.py FAILED\n", "------------------------------------------------------------\n", "Failures:\n", "[1]:\n", " time : 2024-06-01_23:55:26\n", " host : de3bed18af84\n", " rank : 1 (local_rank: 1)\n", " exitcode : 1 (pid: 4361)\n", " error_file: \n", " traceback : To enable traceback see: https://pytorch.org/docs/stable/elastic/errors.html\n", "[2]:\n", " time : 2024-06-01_23:55:26\n", " host : de3bed18af84\n", " rank : 2 (local_rank: 2)\n", " exitcode : 1 (pid: 4362)\n", " error_file: \n", " traceback : To enable traceback see: https://pytorch.org/docs/stable/elastic/errors.html\n", "[3]:\n", " time : 2024-06-01_23:55:26\n", " host : de3bed18af84\n", " rank : 3 (local_rank: 3)\n", " exitcode : 1 (pid: 4363)\n", " error_file: \n", " traceback : To enable traceback see: https://pytorch.org/docs/stable/elastic/errors.html\n", "[4]:\n", " time : 2024-06-01_23:55:26\n", " host : de3bed18af84\n", " rank : 4 (local_rank: 4)\n", " exitcode : 1 (pid: 4364)\n", " error_file: \n", " traceback : To enable traceback see: https://pytorch.org/docs/stable/elastic/errors.html\n", "[5]:\n", " time : 2024-06-01_23:55:26\n", " host : de3bed18af84\n", " rank : 5 (local_rank: 5)\n", " exitcode : 1 (pid: 4365)\n", " error_file: \n", " traceback : To enable traceback see: https://pytorch.org/docs/stable/elastic/errors.html\n", "[6]:\n", " time : 2024-06-01_23:55:26\n", " host : de3bed18af84\n", " rank : 6 (local_rank: 6)\n", " exitcode : 1 (pid: 4366)\n", " error_file: \n", " traceback : To enable traceback see: https://pytorch.org/docs/stable/elastic/errors.html\n", "[7]:\n", " time : 2024-06-01_23:55:26\n", " host : de3bed18af84\n", " rank : 7 (local_rank: 7)\n", " exitcode : 1 (pid: 4367)\n", " error_file: \n", " traceback : To enable traceback see: https://pytorch.org/docs/stable/elastic/errors.html\n", "------------------------------------------------------------\n", "Root Cause (first observed failure):\n", "[0]:\n", " time : 2024-06-01_23:55:26\n", " host : de3bed18af84\n", " rank : 0 (local_rank: 0)\n", " exitcode : 1 (pid: 4360)\n", " error_file: \n", " traceback : To enable traceback see: https://pytorch.org/docs/stable/elastic/errors.html\n", "============================================================\n" ] } ], "source": [ "!torchrun --nproc_per_node 2 --master_port 9527 train_dual.py \\\n", "--workers 8 --device 0,1 --sync-bn --batch 30 --data '/workspace/norway/data.yaml' \\\n", "--img 640 --cfg models/detect/yolov9-c.yaml --weights '' --name yolov9-c --hyp hyp.scratch-high.yaml \\\n", "--min-items 0 --epochs 500 --close-mosaic 15" ] }, { "cell_type": "code", "execution_count": 10, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "[2024-06-02 02:12:23,563] torch.distributed.run: [WARNING] \n", "[2024-06-02 02:12:23,563] torch.distributed.run: [WARNING] *****************************************\n", "[2024-06-02 02:12:23,563] torch.distributed.run: [WARNING] Setting OMP_NUM_THREADS environment variable for each process to be 1 in default, to avoid your system being overloaded, please further tune the variable for optimal performance in your application as needed. \n", "[2024-06-02 02:12:23,563] torch.distributed.run: [WARNING] *****************************************\n", "\u001b[34m\u001b[1mtrain_dual: \u001b[0mweights=, cfg=models/detect/yolov9-c.yaml, data=/workspace/road damage/data.yaml, hyp=hyp.scratch-high.yaml, epochs=500, batch_size=30, imgsz=640, rect=False, resume=False, nosave=False, noval=False, noautoanchor=False, noplots=False, evolve=None, bucket=, cache=None, image_weights=False, device=0,1, multi_scale=False, single_cls=False, optimizer=SGD, sync_bn=True, workers=8, project=runs/train, name=yolov9-c, exist_ok=False, quad=False, cos_lr=False, flat_cos_lr=False, fixed_lr=False, label_smoothing=0.0, patience=100, freeze=[0], save_period=-1, seed=0, local_rank=-1, min_items=0, close_mosaic=15, entity=None, upload_dataset=False, bbox_interval=-1, artifact_alias=latest\n", "YOLO 🚀 v0.1-94-gcd88016 Python-3.10.13 torch-2.2.0 CUDA:0 (NVIDIA GeForce RTX 4090, 24217MiB)\n", " CUDA:1 (NVIDIA GeForce RTX 4090, 24217MiB)\n", "\n", "\u001b[34m\u001b[1mhyperparameters: \u001b[0mlr0=0.01, lrf=0.01, momentum=0.937, weight_decay=0.0005, warmup_epochs=3.0, warmup_momentum=0.8, warmup_bias_lr=0.1, box=7.5, cls=0.5, cls_pw=1.0, obj=0.7, obj_pw=1.0, dfl=1.5, iou_t=0.2, anchor_t=5.0, fl_gamma=0.0, hsv_h=0.015, hsv_s=0.7, hsv_v=0.4, degrees=0.0, translate=0.1, scale=0.9, shear=0.0, perspective=0.0, flipud=0.0, fliplr=0.5, mosaic=1.0, mixup=0.15, copy_paste=0.3\n", "\u001b[34m\u001b[1mClearML: \u001b[0mrun 'pip install clearml' to automatically track, visualize and remotely train YOLO 🚀 in ClearML\n", "\u001b[34m\u001b[1mComet: \u001b[0mrun 'pip install comet_ml' to automatically track and visualize YOLO 🚀 runs in Comet\n", "\u001b[34m\u001b[1mTensorBoard: \u001b[0mStart with 'tensorboard --logdir runs/train', view at http://localhost:6006/\n", "Overriding model.yaml nc=80 with nc=7\n", "\n", " from n params module arguments \n", " 0 -1 1 0 models.common.Silence [] \n", " 1 -1 1 1856 models.common.Conv [3, 64, 3, 2] \n", " 2 -1 1 73984 models.common.Conv [64, 128, 3, 2] \n", " 3 -1 1 291456 models.common.RepNCSPELAN4 [128, 256, 128, 64, 3] \n", " 4 -1 1 164352 models.common.ADown [256, 256] \n", " 5 -1 1 1160448 models.common.RepNCSPELAN4 [256, 512, 256, 128, 3] \n", " 6 -1 1 656384 models.common.ADown [512, 512] \n", " 7 -1 1 4105728 models.common.RepNCSPELAN4 [512, 512, 512, 256, 3] \n", " 8 -1 1 656384 models.common.ADown [512, 512] \n", " 9 -1 1 4105728 models.common.RepNCSPELAN4 [512, 512, 512, 256, 3] \n", " 10 -1 1 656896 models.common.SPPELAN [512, 512, 256] \n", " 11 -1 1 0 torch.nn.modules.upsampling.Upsample [None, 2, 'nearest'] \n", " 12 [-1, 7] 1 0 models.common.Concat [1] \n", " 13 -1 1 3119616 models.common.RepNCSPELAN4 [1024, 512, 512, 256, 1] \n", " 14 -1 1 0 torch.nn.modules.upsampling.Upsample [None, 2, 'nearest'] \n", " 15 [-1, 5] 1 0 models.common.Concat [1] \n", " 16 -1 1 912640 models.common.RepNCSPELAN4 [1024, 256, 256, 128, 1] \n", " 17 -1 1 164352 models.common.ADown [256, 256] \n", " 18 [-1, 13] 1 0 models.common.Concat [1] \n", " 19 -1 1 2988544 models.common.RepNCSPELAN4 [768, 512, 512, 256, 1] \n", " 20 -1 1 656384 models.common.ADown [512, 512] \n", " 21 [-1, 10] 1 0 models.common.Concat [1] \n", " 22 -1 1 3119616 models.common.RepNCSPELAN4 [1024, 512, 512, 256, 1] \n", " 23 5 1 131328 models.common.CBLinear [512, [256]] \n", " 24 7 1 393984 models.common.CBLinear [512, [256, 512]] \n", " 25 9 1 656640 models.common.CBLinear [512, [256, 512, 512]] \n", " 26 0 1 1856 models.common.Conv [3, 64, 3, 2] \n", " 27 -1 1 73984 models.common.Conv [64, 128, 3, 2] \n", " 28 -1 1 212864 models.common.RepNCSPELAN4 [128, 256, 128, 64, 1] \n", " 29 -1 1 164352 models.common.ADown [256, 256] \n", " 30 [23, 24, 25, -1] 1 0 models.common.CBFuse [[0, 0, 0]] \n", " 31 -1 1 847616 models.common.RepNCSPELAN4 [256, 512, 256, 128, 1] \n", " 32 -1 1 656384 models.common.ADown [512, 512] \n", " 33 [24, 25, -1] 1 0 models.common.CBFuse [[1, 1]] \n", " 34 -1 1 2857472 models.common.RepNCSPELAN4 [512, 512, 512, 256, 1] \n", " 35 -1 1 656384 models.common.ADown [512, 512] \n", " 36 [25, -1] 1 0 models.common.CBFuse [[2]] \n", " 37 -1 1 2857472 models.common.RepNCSPELAN4 [512, 512, 512, 256, 1] \n", " 38[31, 34, 37, 16, 19, 22] 1 21556682 models.yolo.DualDDetect [7, [512, 512, 512, 256, 512, 512]]\n", "yolov9-c summary: 1170 layers, 53901386 parameters, 53901354 gradients, 252.0 GFLOPs\n", "\n", "\u001b[34m\u001b[1mAMP: \u001b[0mchecks passed ✅\n", "\u001b[34m\u001b[1moptimizer:\u001b[0m SGD(lr=0.01) with parameter groups 286 weight(decay=0.0), 303 weight(decay=0.00046875), 301 bias\n", "Using SyncBatchNorm()\n", "\u001b[34m\u001b[1malbumentations: \u001b[0mBlur(p=0.01, blur_limit=(3, 7)), MedianBlur(p=0.01, blur_limit=(3, 7)), ToGray(p=0.01), CLAHE(p=0.01, clip_limit=(1, 4.0), tile_grid_size=(8, 8))\n", "\u001b[34m\u001b[1mtrain: \u001b[0mScanning /workspace/road damage/train/labels... 4672 images, 345 backgrou\u001b[0m\n", "\u001b[34m\u001b[1mtrain: \u001b[0mWARNING ⚠️ /workspace/road damage/train/images/Japan_006916_jpg.rf.5dc73c813f7053289379ab9cf4e81173.jpg: 1 duplicate labels removed\n", "\u001b[34m\u001b[1mtrain: \u001b[0mNew cache created: /workspace/road damage/train/labels.cache\n", "\u001b[34m\u001b[1mval: \u001b[0mScanning /workspace/road damage/valid/labels... 450 images, 34 backgrounds,\u001b[0m\n", "\u001b[34m\u001b[1mval: \u001b[0mNew cache created: /workspace/road damage/valid/labels.cache\n", "Plotting labels to runs/train/yolov9-c4/labels.jpg... \n", "Image sizes 640 train, 640 val\n", "Using 16 dataloader workers\n", "Logging results to \u001b[1mruns/train/yolov9-c4\u001b[0m\n", "Starting training for 500 epochs...\n", "\n", " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", " 0/499 20.4G 4.8 6.503 5.217 74 640: Exception in thread Thread-11 (plot_images):\n", "Traceback (most recent call last):\n", " File \"/opt/conda/lib/python3.10/threading.py\", line 1016, in _bootstrap_inner\n", " self.run()\n", " File \"/opt/conda/lib/python3.10/threading.py\", line 953, in run\n", " self._target(*self._args, **self._kwargs)\n", " File \"/workspace/yolov9/utils/plots.py\", line 300, in plot_images\n", " annotator.box_label(box, label, color=color)\n", " File \"/workspace/yolov9/utils/plots.py\", line 86, in box_label\n", " w, h = self.font.getsize(label) # text width, height\n", "AttributeError: 'FreeTypeFont' object has no attribute 'getsize'\n", " 0/499 20.4G 4.903 6.657 5.212 61 640: Exception in thread Thread-12 (plot_images):\n", "Traceback (most recent call last):\n", " File \"/opt/conda/lib/python3.10/threading.py\", line 1016, in _bootstrap_inner\n", " self.run()\n", " File \"/opt/conda/lib/python3.10/threading.py\", line 953, in run\n", " self._target(*self._args, **self._kwargs)\n", " File \"/workspace/yolov9/utils/plots.py\", line 300, in plot_images\n", " annotator.box_label(box, label, color=color)\n", " File \"/workspace/yolov9/utils/plots.py\", line 86, in box_label\n", " w, h = self.font.getsize(label) # text width, height\n", "AttributeError: 'FreeTypeFont' object has no attribute 'getsize'\n", " 0/499 20.4G 4.703 6.511 5.215 57 640: Exception in thread Thread-13 (plot_images):\n", "Traceback (most recent call last):\n", " File \"/opt/conda/lib/python3.10/threading.py\", line 1016, in _bootstrap_inner\n", " self.run()\n", " File \"/opt/conda/lib/python3.10/threading.py\", line 953, in run\n", " self._target(*self._args, **self._kwargs)\n", " File \"/workspace/yolov9/utils/plots.py\", line 300, in plot_images\n", " annotator.box_label(box, label, color=color)\n", " File \"/workspace/yolov9/utils/plots.py\", line 86, in box_label\n", " w, h = self.font.getsize(label) # text width, height\n", "AttributeError: 'FreeTypeFont' object has no attribute 'getsize'\n", " 0/499 20.5G 4.674 6.804 5.251 49 640: ^C\n", "[2024-06-02 02:12:52,085] torch.distributed.elastic.agent.server.api: [WARNING] Received Signals.SIGINT death signal, shutting down workers\n", "[2024-06-02 02:12:52,086] torch.distributed.elastic.multiprocessing.api: [WARNING] Sending process 11363 closing signal SIGINT\n", "[2024-06-02 02:12:52,086] torch.distributed.elastic.multiprocessing.api: [WARNING] Sending process 11364 closing signal SIGINT\n", " 0/499 20.5G 4.674 6.804 5.251 49 640: \n", "Traceback (most recent call last):\n", " File \"/workspace/yolov9/train_dual.py\", line 644, in \n", " main(opt)\n", " File \"/workspace/yolov9/train_dual.py\", line 538, in main\n", " train(opt.hyp, opt, device, callbacks)\n", " File \"/workspace/yolov9/train_dual.py\", line 314, in train\n", " pred = model(imgs) # forward\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1511, in _wrapped_call_impl\n", " return self._call_impl(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1520, in _call_impl\n", " return forward_call(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/parallel/distributed.py\", line 1523, in forward\n", " else self._run_ddp_forward(*inputs, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/parallel/distributed.py\", line 1359, in _run_ddp_forward\n", " return self.module(*inputs, **kwargs) # type: ignore[index]\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1511, in _wrapped_call_impl\n", " return self._call_impl(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1520, in _call_impl\n", " return forward_call(*args, **kwargs)\n", " File \"/workspace/yolov9/models/yolo.py\", line 633, in forward\n", " return self._forward_once(x, profile, visualize) # single-scale inference, train\n", " File \"/workspace/yolov9/models/yolo.py\", line 533, in _forward_once\n", " x = m(x) # run\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1511, in _wrapped_call_impl\n", " return self._call_impl(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1520, in _call_impl\n", " return forward_call(*args, **kwargs)\n", " File \"/workspace/yolov9/models/yolo.py\", line 225, in forward\n", " d1.append(torch.cat((self.cv2[i](x[i]), self.cv3[i](x[i])), 1))\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1511, in _wrapped_call_impl\n", " return self._call_impl(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1520, in _call_impl\n", " return forward_call(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/container.py\", line 217, in forward\n", " input = module(input)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1511, in _wrapped_call_impl\n", " return self._call_impl(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1520, in _call_impl\n", " return forward_call(*args, **kwargs)\n", " File \"/workspace/yolov9/models/common.py\", line 54, in forward\n", " return self.act(self.bn(self.conv(x)))\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1511, in _wrapped_call_impl\n", " return self._call_impl(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1520, in _call_impl\n", " return forward_call(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/batchnorm.py\", line 767, in forward\n", " return sync_batch_norm.apply(\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/autograd/function.py\", line 553, in apply\n", " return super().apply(*args, **kwargs) # type: ignore[misc]\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/_functions.py\", line 89, in forward\n", " counts = count_all.view(-1)\n", "KeyboardInterrupt\n", "Traceback (most recent call last):\n", " File \"/workspace/yolov9/train_dual.py\", line 644, in \n", " main(opt)\n", " File \"/workspace/yolov9/train_dual.py\", line 538, in main\n", " train(opt.hyp, opt, device, callbacks)\n", " File \"/workspace/yolov9/train_dual.py\", line 314, in train\n", " pred = model(imgs) # forward\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1511, in _wrapped_call_impl\n", " return self._call_impl(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1520, in _call_impl\n", " return forward_call(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/parallel/distributed.py\", line 1523, in forward\n", " else self._run_ddp_forward(*inputs, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/parallel/distributed.py\", line 1359, in _run_ddp_forward\n", " return self.module(*inputs, **kwargs) # type: ignore[index]\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1511, in _wrapped_call_impl\n", " return self._call_impl(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1520, in _call_impl\n", " return forward_call(*args, **kwargs)\n", " File \"/workspace/yolov9/models/yolo.py\", line 633, in forward\n", " return self._forward_once(x, profile, visualize) # single-scale inference, train\n", " File \"/workspace/yolov9/models/yolo.py\", line 533, in _forward_once\n", " x = m(x) # run\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1511, in _wrapped_call_impl\n", " return self._call_impl(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1520, in _call_impl\n", " return forward_call(*args, **kwargs)\n", " File \"/workspace/yolov9/models/yolo.py\", line 225, in forward\n", " d1.append(torch.cat((self.cv2[i](x[i]), self.cv3[i](x[i])), 1))\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1511, in _wrapped_call_impl\n", " return self._call_impl(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1520, in _call_impl\n", " return forward_call(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/container.py\", line 217, in forward\n", " input = module(input)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1511, in _wrapped_call_impl\n", " return self._call_impl(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1520, in _call_impl\n", " return forward_call(*args, **kwargs)\n", " File \"/workspace/yolov9/models/common.py\", line 54, in forward\n", " return self.act(self.bn(self.conv(x)))\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1511, in _wrapped_call_impl\n", " return self._call_impl(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py\", line 1520, in _call_impl\n", " return forward_call(*args, **kwargs)\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/batchnorm.py\", line 767, in forward\n", " return sync_batch_norm.apply(\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/autograd/function.py\", line 553, in apply\n", " return super().apply(*args, **kwargs) # type: ignore[misc]\n", " File \"/opt/conda/lib/python3.10/site-packages/torch/nn/modules/_functions.py\", line 89, in forward\n", " counts = count_all.view(-1)\n", "KeyboardInterrupt\n" ] } ], "source": [ "!torchrun --nproc_per_node 2 --master_port 9527 train_dual.py \\\n", "--workers 8 --device 0,1 --sync-bn --batch 30 --data '/workspace/road damage/data.yaml' \\\n", "--img 640 --cfg models/detect/yolov9-c.yaml --weights '' --name yolov9-c --hyp hyp.scratch-high.yaml \\\n", "--min-items 0 --epochs 500 --close-mosaic 15" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "!torchrun --nproc_per_node 2 --master_port 9527 train_dual.py \\\n", "--workers 8 --device 0,1 --sync-bn --batch 30 --data '/workspace/road damage/data.yaml' \\\n", "--img 640 --cfg models/detect/yolov9-c.yaml --weights '' --name yolov9-c --hyp hyp.scratch-high.yaml \\\n", "--min-items 0 --epochs 500 --close-mosaic 15" ] } ], "metadata": { "accelerator": "GPU", "colab": { "gpuType": "L4", "machine_shape": "hm", "provenance": [] }, "kernelspec": { "display_name": "Python 3 (ipykernel)", "language": "python", "name": "python3" }, "language_info": { "codemirror_mode": { "name": "ipython", "version": 3 }, "file_extension": ".py", "mimetype": "text/x-python", "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", "version": "3.10.13" } }, "nbformat": 4, "nbformat_minor": 4 }