es_project.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "name": "es_project.ipynb",
      "provenance": [],
      "collapsed_sections": [],
      "mount_file_id": "1wlUCPxr_HBUpXwVJtlfFQ8j-ObJfK8BK",
      "authorship_tag": "ABX9TyMHokZRuSH6OqPUy8dd/MCg",
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    },
    "accelerator": "GPU"
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/Indrajith1446/Compiler_Lab/blob/main/es_project.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "0EoW65SbieGP",
        "outputId": "364e34b4-a384-4895-c773-f53d56abad7a"
      },
      "source": [
        "!git clone https://github.com/ultralytics/yolov5"
      ],
      "execution_count": null,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Cloning into 'yolov5'...\n",
            "remote: Enumerating objects: 10081, done.\u001b[K\n",
            "remote: Total 10081 (delta 0), reused 0 (delta 0), pack-reused 10081\u001b[K\n",
            "Receiving objects: 100% (10081/10081), 10.36 MiB | 23.17 MiB/s, done.\n",
            "Resolving deltas: 100% (6994/6994), done.\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 1000
        },
        "id": "tmMgsNYbitck",
        "outputId": "460731ad-e5d3-467f-82e5-c7bbc697ef8c"
      },
      "source": [
        "pip install -U -r yolov5/requirements.txt"
      ],
      "execution_count": null,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Requirement already satisfied: matplotlib>=3.2.2 in /usr/local/lib/python3.7/dist-packages (from -r yolov5/requirements.txt (line 4)) (3.2.2)\n",
            "Collecting matplotlib>=3.2.2\n",
            "  Downloading matplotlib-3.5.0-cp37-cp37m-manylinux_2_5_x86_64.manylinux1_x86_64.whl (11.2 MB)\n",
            "\u001b[K     |████████████████████████████████| 11.2 MB 5.4 MB/s \n",
            "\u001b[?25hRequirement already satisfied: numpy>=1.18.5 in /usr/local/lib/python3.7/dist-packages (from -r yolov5/requirements.txt (line 5)) (1.19.5)\n",
            "Collecting numpy>=1.18.5\n",
            "  Downloading numpy-1.21.4-cp37-cp37m-manylinux_2_12_x86_64.manylinux2010_x86_64.whl (15.7 MB)\n",
            "\u001b[K     |████████████████████████████████| 15.7 MB 34.2 MB/s \n",
            "\u001b[?25hRequirement already satisfied: opencv-python>=4.1.2 in /usr/local/lib/python3.7/dist-packages (from -r yolov5/requirements.txt (line 6)) (4.1.2.30)\n",
            "Collecting opencv-python>=4.1.2\n",
            "  Downloading opencv_python-4.5.4.60-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (60.3 MB)\n",
            "\u001b[K     |████████████████████████████████| 60.3 MB 53 kB/s \n",
            "\u001b[?25hRequirement already satisfied: Pillow>=7.1.2 in /usr/local/lib/python3.7/dist-packages (from -r yolov5/requirements.txt (line 7)) (7.1.2)\n",
            "Collecting Pillow>=7.1.2\n",
            "  Downloading Pillow-8.4.0-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (3.1 MB)\n",
            "\u001b[K     |████████████████████████████████| 3.1 MB 20.9 MB/s \n",
            "\u001b[?25hCollecting PyYAML>=5.3.1\n",
            "  Downloading PyYAML-6.0-cp37-cp37m-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl (596 kB)\n",
            "\u001b[K     |████████████████████████████████| 596 kB 40.0 MB/s \n",
            "\u001b[?25hRequirement already satisfied: requests>=2.23.0 in /usr/local/lib/python3.7/dist-packages (from -r yolov5/requirements.txt (line 9)) (2.23.0)\n",
            "Collecting requests>=2.23.0\n",
            "  Downloading requests-2.26.0-py2.py3-none-any.whl (62 kB)\n",
            "\u001b[K     |████████████████████████████████| 62 kB 763 kB/s \n",
            "\u001b[?25hRequirement already satisfied: scipy>=1.4.1 in /usr/local/lib/python3.7/dist-packages (from -r yolov5/requirements.txt (line 10)) (1.4.1)\n",
            "Collecting scipy>=1.4.1\n",
            "  Downloading scipy-1.7.3-cp37-cp37m-manylinux_2_12_x86_64.manylinux2010_x86_64.whl (38.1 MB)\n",
            "\u001b[K     |████████████████████████████████| 38.1 MB 1.4 MB/s \n",
            "\u001b[?25hRequirement already satisfied: torch>=1.7.0 in /usr/local/lib/python3.7/dist-packages (from -r yolov5/requirements.txt (line 11)) (1.10.0+cu111)\n",
            "Requirement already satisfied: torchvision>=0.8.1 in /usr/local/lib/python3.7/dist-packages (from -r yolov5/requirements.txt (line 12)) (0.11.1+cu111)\n",
            "Requirement already satisfied: tqdm>=4.41.0 in /usr/local/lib/python3.7/dist-packages (from -r yolov5/requirements.txt (line 13)) (4.62.3)\n",
            "Requirement already satisfied: tensorboard>=2.4.1 in /usr/local/lib/python3.7/dist-packages (from -r yolov5/requirements.txt (line 16)) (2.7.0)\n",
            "Requirement already satisfied: pandas>=1.1.4 in /usr/local/lib/python3.7/dist-packages (from -r yolov5/requirements.txt (line 20)) (1.1.5)\n",
            "Collecting pandas>=1.1.4\n",
            "  Downloading pandas-1.3.4-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (11.3 MB)\n",
            "\u001b[K     |████████████████████████████████| 11.3 MB 35.0 MB/s \n",
            "\u001b[?25hRequirement already satisfied: seaborn>=0.11.0 in /usr/local/lib/python3.7/dist-packages (from -r yolov5/requirements.txt (line 21)) (0.11.2)\n",
            "Collecting thop\n",
            "  Downloading thop-0.0.31.post2005241907-py3-none-any.whl (8.7 kB)\n",
            "Requirement already satisfied: pyparsing>=2.2.1 in /usr/local/lib/python3.7/dist-packages (from matplotlib>=3.2.2->-r yolov5/requirements.txt (line 4)) (3.0.6)\n",
            "Requirement already satisfied: setuptools-scm>=4 in /usr/local/lib/python3.7/dist-packages (from matplotlib>=3.2.2->-r yolov5/requirements.txt (line 4)) (6.3.2)\n",
            "Requirement already satisfied: kiwisolver>=1.0.1 in /usr/local/lib/python3.7/dist-packages (from matplotlib>=3.2.2->-r yolov5/requirements.txt (line 4)) (1.3.2)\n",
            "Requirement already satisfied: packaging>=20.0 in /usr/local/lib/python3.7/dist-packages (from matplotlib>=3.2.2->-r yolov5/requirements.txt (line 4)) (21.3)\n",
            "Requirement already satisfied: cycler>=0.10 in /usr/local/lib/python3.7/dist-packages (from matplotlib>=3.2.2->-r yolov5/requirements.txt (line 4)) (0.11.0)\n",
            "Requirement already satisfied: python-dateutil>=2.7 in /usr/local/lib/python3.7/dist-packages (from matplotlib>=3.2.2->-r yolov5/requirements.txt (line 4)) (2.8.2)\n",
            "Requirement already satisfied: fonttools>=4.22.0 in /usr/local/lib/python3.7/dist-packages (from matplotlib>=3.2.2->-r yolov5/requirements.txt (line 4)) (4.28.1)\n",
            "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.7/dist-packages (from requests>=2.23.0->-r yolov5/requirements.txt (line 9)) (2021.10.8)\n",
            "Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.7/dist-packages (from requests>=2.23.0->-r yolov5/requirements.txt (line 9)) (2.10)\n",
            "Requirement already satisfied: charset-normalizer~=2.0.0 in /usr/local/lib/python3.7/dist-packages (from requests>=2.23.0->-r yolov5/requirements.txt (line 9)) (2.0.7)\n",
            "Requirement already satisfied: urllib3<1.27,>=1.21.1 in /usr/local/lib/python3.7/dist-packages (from requests>=2.23.0->-r yolov5/requirements.txt (line 9)) (1.24.3)\n",
            "Requirement already satisfied: typing-extensions in /usr/local/lib/python3.7/dist-packages (from torch>=1.7.0->-r yolov5/requirements.txt (line 11)) (3.10.0.2)\n",
            "Requirement already satisfied: werkzeug>=0.11.15 in /usr/local/lib/python3.7/dist-packages (from tensorboard>=2.4.1->-r yolov5/requirements.txt (line 16)) (1.0.1)\n",
            "Requirement already satisfied: markdown>=2.6.8 in /usr/local/lib/python3.7/dist-packages (from tensorboard>=2.4.1->-r yolov5/requirements.txt (line 16)) (3.3.6)\n",
            "Requirement already satisfied: wheel>=0.26 in /usr/local/lib/python3.7/dist-packages (from tensorboard>=2.4.1->-r yolov5/requirements.txt (line 16)) (0.37.0)\n",
            "Requirement already satisfied: tensorboard-data-server<0.7.0,>=0.6.0 in /usr/local/lib/python3.7/dist-packages (from tensorboard>=2.4.1->-r yolov5/requirements.txt (line 16)) (0.6.1)\n",
            "Requirement already satisfied: absl-py>=0.4 in /usr/local/lib/python3.7/dist-packages (from tensorboard>=2.4.1->-r yolov5/requirements.txt (line 16)) (0.12.0)\n",
            "Requirement already satisfied: grpcio>=1.24.3 in /usr/local/lib/python3.7/dist-packages (from tensorboard>=2.4.1->-r yolov5/requirements.txt (line 16)) (1.42.0)\n",
            "Requirement already satisfied: protobuf>=3.6.0 in /usr/local/lib/python3.7/dist-packages (from tensorboard>=2.4.1->-r yolov5/requirements.txt (line 16)) (3.17.3)\n",
            "Requirement already satisfied: google-auth<3,>=1.6.3 in /usr/local/lib/python3.7/dist-packages (from tensorboard>=2.4.1->-r yolov5/requirements.txt (line 16)) (1.35.0)\n",
            "Requirement already satisfied: tensorboard-plugin-wit>=1.6.0 in /usr/local/lib/python3.7/dist-packages (from tensorboard>=2.4.1->-r yolov5/requirements.txt (line 16)) (1.8.0)\n",
            "Requirement already satisfied: google-auth-oauthlib<0.5,>=0.4.1 in /usr/local/lib/python3.7/dist-packages (from tensorboard>=2.4.1->-r yolov5/requirements.txt (line 16)) (0.4.6)\n",
            "Requirement already satisfied: setuptools>=41.0.0 in /usr/local/lib/python3.7/dist-packages (from tensorboard>=2.4.1->-r yolov5/requirements.txt (line 16)) (57.4.0)\n",
            "Requirement already satisfied: pytz>=2017.3 in /usr/local/lib/python3.7/dist-packages (from pandas>=1.1.4->-r yolov5/requirements.txt (line 20)) (2018.9)\n",
            "Requirement already satisfied: six in /usr/local/lib/python3.7/dist-packages (from absl-py>=0.4->tensorboard>=2.4.1->-r yolov5/requirements.txt (line 16)) (1.15.0)\n",
            "Requirement already satisfied: pyasn1-modules>=0.2.1 in /usr/local/lib/python3.7/dist-packages (from google-auth<3,>=1.6.3->tensorboard>=2.4.1->-r yolov5/requirements.txt (line 16)) (0.2.8)\n",
            "Requirement already satisfied: cachetools<5.0,>=2.0.0 in /usr/local/lib/python3.7/dist-packages (from google-auth<3,>=1.6.3->tensorboard>=2.4.1->-r yolov5/requirements.txt (line 16)) (4.2.4)\n",
            "Requirement already satisfied: rsa<5,>=3.1.4 in /usr/local/lib/python3.7/dist-packages (from google-auth<3,>=1.6.3->tensorboard>=2.4.1->-r yolov5/requirements.txt (line 16)) (4.7.2)\n",
            "Requirement already satisfied: requests-oauthlib>=0.7.0 in /usr/local/lib/python3.7/dist-packages (from google-auth-oauthlib<0.5,>=0.4.1->tensorboard>=2.4.1->-r yolov5/requirements.txt (line 16)) (1.3.0)\n",
            "Requirement already satisfied: importlib-metadata>=4.4 in /usr/local/lib/python3.7/dist-packages (from markdown>=2.6.8->tensorboard>=2.4.1->-r yolov5/requirements.txt (line 16)) (4.8.2)\n",
            "Requirement already satisfied: zipp>=0.5 in /usr/local/lib/python3.7/dist-packages (from importlib-metadata>=4.4->markdown>=2.6.8->tensorboard>=2.4.1->-r yolov5/requirements.txt (line 16)) (3.6.0)\n",
            "Requirement already satisfied: pyasn1<0.5.0,>=0.4.6 in /usr/local/lib/python3.7/dist-packages (from pyasn1-modules>=0.2.1->google-auth<3,>=1.6.3->tensorboard>=2.4.1->-r yolov5/requirements.txt (line 16)) (0.4.8)\n",
            "Requirement already satisfied: oauthlib>=3.0.0 in /usr/local/lib/python3.7/dist-packages (from requests-oauthlib>=0.7.0->google-auth-oauthlib<0.5,>=0.4.1->tensorboard>=2.4.1->-r yolov5/requirements.txt (line 16)) (3.1.1)\n",
            "Requirement already satisfied: tomli>=1.0.0 in /usr/local/lib/python3.7/dist-packages (from setuptools-scm>=4->matplotlib>=3.2.2->-r yolov5/requirements.txt (line 4)) (1.2.2)\n",
            "Installing collected packages: requests, Pillow, numpy, scipy, pandas, matplotlib, thop, PyYAML, opencv-python\n",
            "  Attempting uninstall: requests\n",
            "    Found existing installation: requests 2.23.0\n",
            "    Uninstalling requests-2.23.0:\n",
            "      Successfully uninstalled requests-2.23.0\n",
            "  Attempting uninstall: Pillow\n",
            "    Found existing installation: Pillow 7.1.2\n",
            "    Uninstalling Pillow-7.1.2:\n",
            "      Successfully uninstalled Pillow-7.1.2\n",
            "  Attempting uninstall: numpy\n",
            "    Found existing installation: numpy 1.19.5\n",
            "    Uninstalling numpy-1.19.5:\n",
            "      Successfully uninstalled numpy-1.19.5\n",
            "  Attempting uninstall: scipy\n",
            "    Found existing installation: scipy 1.4.1\n",
            "    Uninstalling scipy-1.4.1:\n",
            "      Successfully uninstalled scipy-1.4.1\n",
            "  Attempting uninstall: pandas\n",
            "    Found existing installation: pandas 1.1.5\n",
            "    Uninstalling pandas-1.1.5:\n",
            "      Successfully uninstalled pandas-1.1.5\n",
            "  Attempting uninstall: matplotlib\n",
            "    Found existing installation: matplotlib 3.2.2\n",
            "    Uninstalling matplotlib-3.2.2:\n",
            "      Successfully uninstalled matplotlib-3.2.2\n",
            "  Attempting uninstall: PyYAML\n",
            "    Found existing installation: PyYAML 3.13\n",
            "    Uninstalling PyYAML-3.13:\n",
            "      Successfully uninstalled PyYAML-3.13\n",
            "  Attempting uninstall: opencv-python\n",
            "    Found existing installation: opencv-python 4.1.2.30\n",
            "    Uninstalling opencv-python-4.1.2.30:\n",
            "      Successfully uninstalled opencv-python-4.1.2.30\n",
            "\u001b[31mERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.\n",
            "yellowbrick 1.3.post1 requires numpy<1.20,>=1.16.0, but you have numpy 1.21.4 which is incompatible.\n",
            "google-colab 1.0.0 requires pandas~=1.1.0; python_version >= \"3.0\", but you have pandas 1.3.4 which is incompatible.\n",
            "google-colab 1.0.0 requires requests~=2.23.0, but you have requests 2.26.0 which is incompatible.\n",
            "datascience 0.10.6 requires folium==0.2.1, but you have folium 0.8.3 which is incompatible.\n",
            "albumentations 0.1.12 requires imgaug<0.2.7,>=0.2.5, but you have imgaug 0.2.9 which is incompatible.\u001b[0m\n",
            "Successfully installed Pillow-8.4.0 PyYAML-6.0 matplotlib-3.5.0 numpy-1.21.4 opencv-python-4.5.4.60 pandas-1.3.4 requests-2.26.0 scipy-1.7.3 thop-0.0.31.post2005241907\n"
          ]
        },
        {
          "output_type": "display_data",
          "data": {
            "application/vnd.colab-display-data+json": {
              "pip_warning": {
                "packages": [
                  "PIL",
                  "matplotlib",
                  "mpl_toolkits",
                  "numpy",
                  "pandas"
                ]
              }
            }
          },
          "metadata": {}
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "SU-GPZpWK-xv",
        "outputId": "ff4a08a7-06b2-4173-8f6e-e8f2600de513"
      },
      "source": [
        "from google.colab import drive\n",
        "drive.mount('/content/drive')"
      ],
      "execution_count": null,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Drive already mounted at /content/drive; to attempt to forcibly remount, call drive.mount(\"/content/drive\", force_remount=True).\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "LpcErcuHi-tm",
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 374
        },
        "outputId": "b4f98254-3e0c-44dd-b3a9-7ff57b60571f"
      },
      "source": [
        "import os\n",
        "import re\n",
        "\n",
        "for folder in ['train', 'valid']:\n",
        "    file_list = os.listdir(f'./drive/MyDrive/dataset/{folder}')\n",
        "    for file in file_list:\n",
        "        if not file.endswith('.txt'):\n",
        "            continue\n",
        "        with open(f'./drive/MyDrive/dataset/{folder}/{file}', 'r+') as f:\n",
        "            lines = f.read()\n",
        "            replaced = re.sub(r'0(?=\\s\\d+\\.\\d+){4}', '1', lines)\n",
        "            replaced = re.sub(r'2(?=\\s\\d+\\.\\d+){4}', '0', lines)\n",
        "            f.seek(0)\n",
        "            f.write(replaced)\n",
        "            f.truncate()"
      ],
      "execution_count": null,
      "outputs": [
        {
          "output_type": "error",
          "ename": "KeyboardInterrupt",
          "evalue": "ignored",
          "traceback": [
            "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
            "\u001b[0;31mKeyboardInterrupt\u001b[0m                         Traceback (most recent call last)",
            "\u001b[0;32m<ipython-input-6-71d474335189>\u001b[0m in \u001b[0;36m<module>\u001b[0;34m()\u001b[0m\n\u001b[1;32m      8\u001b[0m             \u001b[0;32mcontinue\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m      9\u001b[0m         \u001b[0;32mwith\u001b[0m \u001b[0mopen\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34mf'./drive/MyDrive/dataset/{folder}/{file}'\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0;34m'r+'\u001b[0m\u001b[0;34m)\u001b[0m \u001b[0;32mas\u001b[0m \u001b[0mf\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m---> 10\u001b[0;31m             \u001b[0mlines\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mf\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mread\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m     11\u001b[0m             \u001b[0mreplaced\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mre\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0msub\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34mr'0(?=\\s\\d+\\.\\d+){4}'\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0;34m'1'\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mlines\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m     12\u001b[0m             \u001b[0mreplaced\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mre\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0msub\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34mr'2(?=\\s\\d+\\.\\d+){4}'\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0;34m'0'\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mlines\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
            "\u001b[0;32m/usr/lib/python3.7/codecs.py\u001b[0m in \u001b[0;36mdecode\u001b[0;34m(self, input, final)\u001b[0m\n\u001b[1;32m    317\u001b[0m         \u001b[0;32mraise\u001b[0m \u001b[0mNotImplementedError\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    318\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 319\u001b[0;31m     \u001b[0;32mdef\u001b[0m \u001b[0mdecode\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mself\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0minput\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mfinal\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0;32mFalse\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    320\u001b[0m         \u001b[0;31m# decode input (taking the buffer into account)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    321\u001b[0m         \u001b[0mdata\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mself\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mbuffer\u001b[0m \u001b[0;34m+\u001b[0m \u001b[0minput\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
            "\u001b[0;31mKeyboardInterrupt\u001b[0m: "
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "ptr32t3ZGOGR",
        "outputId": "cc9f6ec0-c783-45ab-8389-4a7ad61d6f87"
      },
      "source": [
        "!python yolov5/train.py --img 416 --batch 16 --epochs 150 --data data.yaml --cfg yolov5s.yaml --weights ‘’ --name robo4_epoch150_s --adam "
      ],
      "execution_count": 3,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Downloading https://ultralytics.com/assets/Arial.ttf to /root/.config/Ultralytics/Arial.ttf...\n",
            "\u001b[34m\u001b[1mtrain: \u001b[0mweights=‘’, cfg=yolov5s.yaml, data=data.yaml, hyp=yolov5/data/hyps/hyp.scratch.yaml, epochs=150, batch_size=16, imgsz=416, rect=False, resume=False, nosave=False, noval=False, noautoanchor=False, evolve=None, bucket=, cache=None, image_weights=False, device=, multi_scale=False, single_cls=False, adam=True, sync_bn=False, workers=8, project=yolov5/runs/train, name=robo4_epoch150_s, exist_ok=False, quad=False, linear_lr=False, label_smoothing=0.0, patience=100, freeze=0, save_period=-1, local_rank=-1, entity=None, upload_dataset=False, bbox_interval=-1, artifact_alias=latest\n",
            "\u001b[34m\u001b[1mgithub: \u001b[0mup to date with https://github.com/ultralytics/yolov5 ✅\n",
            "YOLOv5 🚀 v6.0-113-g5ca5dd4 torch 1.10.0+cu111 CUDA:0 (Tesla K80, 11441MiB)\n",
            "\n",
            "\u001b[34m\u001b[1mhyperparameters: \u001b[0mlr0=0.01, lrf=0.1, momentum=0.937, weight_decay=0.0005, warmup_epochs=3.0, warmup_momentum=0.8, warmup_bias_lr=0.1, box=0.05, cls=0.5, cls_pw=1.0, obj=1.0, obj_pw=1.0, iou_t=0.2, anchor_t=4.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.5, shear=0.0, perspective=0.0, flipud=0.0, fliplr=0.5, mosaic=1.0, mixup=0.0, copy_paste=0.0\n",
            "\u001b[34m\u001b[1mWeights & Biases: \u001b[0mrun 'pip install wandb' to automatically track and visualize YOLOv5 🚀 runs (RECOMMENDED)\n",
            "\u001b[34m\u001b[1mTensorBoard: \u001b[0mStart with 'tensorboard --logdir yolov5/runs/train', view at http://localhost:6006/\n",
            "Overriding model.yaml nc=80 with nc=2\n",
            "\n",
            "                 from  n    params  module                                  arguments                     \n",
            "  0                -1  1      3520  models.common.Conv                      [3, 32, 6, 2, 2]              \n",
            "  1                -1  1     18560  models.common.Conv                      [32, 64, 3, 2]                \n",
            "  2                -1  1     18816  models.common.C3                        [64, 64, 1]                   \n",
            "  3                -1  1     73984  models.common.Conv                      [64, 128, 3, 2]               \n",
            "  4                -1  2    115712  models.common.C3                        [128, 128, 2]                 \n",
            "  5                -1  1    295424  models.common.Conv                      [128, 256, 3, 2]              \n",
            "  6                -1  3    625152  models.common.C3                        [256, 256, 3]                 \n",
            "  7                -1  1   1180672  models.common.Conv                      [256, 512, 3, 2]              \n",
            "  8                -1  1   1182720  models.common.C3                        [512, 512, 1]                 \n",
            "  9                -1  1    656896  models.common.SPPF                      [512, 512, 5]                 \n",
            " 10                -1  1    131584  models.common.Conv                      [512, 256, 1, 1]              \n",
            " 11                -1  1         0  torch.nn.modules.upsampling.Upsample    [None, 2, 'nearest']          \n",
            " 12           [-1, 6]  1         0  models.common.Concat                    [1]                           \n",
            " 13                -1  1    361984  models.common.C3                        [512, 256, 1, False]          \n",
            " 14                -1  1     33024  models.common.Conv                      [256, 128, 1, 1]              \n",
            " 15                -1  1         0  torch.nn.modules.upsampling.Upsample    [None, 2, 'nearest']          \n",
            " 16           [-1, 4]  1         0  models.common.Concat                    [1]                           \n",
            " 17                -1  1     90880  models.common.C3                        [256, 128, 1, False]          \n",
            " 18                -1  1    147712  models.common.Conv                      [128, 128, 3, 2]              \n",
            " 19          [-1, 14]  1         0  models.common.Concat                    [1]                           \n",
            " 20                -1  1    296448  models.common.C3                        [256, 256, 1, False]          \n",
            " 21                -1  1    590336  models.common.Conv                      [256, 256, 3, 2]              \n",
            " 22          [-1, 10]  1         0  models.common.Concat                    [1]                           \n",
            " 23                -1  1   1182720  models.common.C3                        [512, 512, 1, False]          \n",
            " 24      [17, 20, 23]  1     18879  models.yolo.Detect                      [2, [[10, 13, 16, 30, 33, 23], [30, 61, 62, 45, 59, 119], [116, 90, 156, 198, 373, 326]], [128, 256, 512]]\n",
            "Model Summary: 270 layers, 7025023 parameters, 7025023 gradients, 15.9 GFLOPs\n",
            "\n",
            "Scaled weight_decay = 0.0005\n",
            "\u001b[34m\u001b[1moptimizer:\u001b[0m Adam with parameter groups 57 weight, 60 weight (no decay), 60 bias\n",
            "\u001b[34m\u001b[1malbumentations: \u001b[0mversion 1.0.3 required by YOLOv5, but version 0.1.12 is currently installed\n",
            "\u001b[34m\u001b[1mtrain: \u001b[0mScanning 'drive/MyDrive/dataset2/train' images and labels...2034 found, 0 missing, 0 empty, 0 corrupted: 100% 2034/2034 [18:29<00:00,  1.83it/s]\n",
            "\u001b[34m\u001b[1mtrain: \u001b[0mNew cache created: drive/MyDrive/dataset2/train.cache\n",
            "\u001b[34m\u001b[1mval: \u001b[0mScanning 'drive/MyDrive/dataset2/valid' images and labels...170 found, 0 missing, 0 empty, 0 corrupted: 100% 170/170 [01:27<00:00,  1.95it/s]\n",
            "\u001b[34m\u001b[1mval: \u001b[0mNew cache created: drive/MyDrive/dataset2/valid.cache\n",
            "Plotting labels to yolov5/runs/train/robo4_epoch150_s/labels.jpg... \n",
            "\n",
            "\u001b[34m\u001b[1mAutoAnchor: \u001b[0m5.56 anchors/target, 1.000 Best Possible Recall (BPR). Current anchors are a good fit to dataset ✅\n",
            "Image sizes 416 train, 416 val\n",
            "Using 2 dataloader workers\n",
            "Logging results to \u001b[1myolov5/runs/train/robo4_epoch150_s\u001b[0m\n",
            "Starting training for 150 epochs...\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     0/149     1.48G   0.08816   0.05031   0.01849         6       416: 100% 128/128 [01:20<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.52it/s]\n",
            "                 all        170       1004     0.0671     0.0591     0.0228    0.00489\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     1/149     1.69G   0.06982   0.04356   0.01651         7       416: 100% 128/128 [01:14<00:00,  1.73it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.03it/s]\n",
            "                 all        170       1004      0.685      0.163      0.129     0.0262\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     2/149     1.69G   0.06207   0.04141   0.01524         5       416: 100% 128/128 [01:13<00:00,  1.74it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.19it/s]\n",
            "                 all        170       1004      0.791      0.241      0.262     0.0788\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     3/149     1.69G   0.05896   0.04025   0.01445        15       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.32it/s]\n",
            "                 all        170       1004      0.483      0.486      0.458      0.174\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     4/149     1.69G   0.05708   0.03736   0.01323         8       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.44it/s]\n",
            "                 all        170       1004      0.568      0.431      0.415      0.145\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     5/149     1.69G   0.05552   0.03818   0.01232        12       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:03<00:00,  1.95it/s]\n",
            "                 all        170       1004      0.388      0.472      0.387      0.143\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     6/149     1.69G   0.05404   0.03585   0.01187        21       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.38it/s]\n",
            "                 all        170       1004      0.628       0.46      0.467      0.153\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     7/149     1.69G    0.0537    0.0367   0.01155        10       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.32it/s]\n",
            "                 all        170       1004      0.582      0.507      0.508      0.182\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     8/149     1.69G   0.05188   0.03577   0.01133        12       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.28it/s]\n",
            "                 all        170       1004      0.696      0.551      0.572      0.212\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     9/149     1.69G   0.05134   0.03527   0.01062        10       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.42it/s]\n",
            "                 all        170       1004      0.694      0.563      0.585      0.238\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    10/149     1.69G   0.04882   0.03438   0.01026        12       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.44it/s]\n",
            "                 all        170       1004      0.732      0.577      0.621      0.236\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    11/149     1.69G   0.04902   0.03368  0.009781         8       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.33it/s]\n",
            "                 all        170       1004      0.788      0.621      0.659      0.255\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    12/149     1.69G   0.04874   0.03411  0.009416         7       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.37it/s]\n",
            "                 all        170       1004      0.699      0.605      0.623      0.245\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    13/149     1.69G    0.0468   0.03285  0.009244        12       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.29it/s]\n",
            "                 all        170       1004      0.771      0.627      0.674      0.272\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    14/149     1.69G   0.04776   0.03401  0.008985        12       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.40it/s]\n",
            "                 all        170       1004      0.665      0.498       0.53      0.192\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    15/149     1.69G   0.04801    0.0329  0.009311         8       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.18it/s]\n",
            "                 all        170       1004      0.817      0.641      0.705      0.332\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    16/149     1.69G   0.04776   0.03214  0.008762        17       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.39it/s]\n",
            "                 all        170       1004      0.757      0.616      0.657      0.228\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    17/149     1.69G   0.04636   0.03219  0.008602        16       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.42it/s]\n",
            "                 all        170       1004      0.721      0.582      0.615      0.265\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    18/149     1.69G   0.04577   0.03247  0.008166         9       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.45it/s]\n",
            "                 all        170       1004      0.818      0.611      0.689      0.317\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    19/149     1.69G   0.04595   0.03212  0.008728         5       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.36it/s]\n",
            "                 all        170       1004      0.694      0.621      0.632      0.275\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    20/149     1.69G   0.04519   0.03249  0.008134        12       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.26it/s]\n",
            "                 all        170       1004      0.844       0.65      0.701      0.289\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    21/149     1.69G   0.04443   0.03205   0.00766        11       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.44it/s]\n",
            "                 all        170       1004      0.839      0.616      0.685      0.306\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    22/149     1.69G   0.04489   0.03199  0.007976        19       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.36it/s]\n",
            "                 all        170       1004      0.762      0.622      0.692      0.346\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    23/149     1.69G   0.04523   0.03225   0.00778        11       416: 100% 128/128 [01:12<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.36it/s]\n",
            "                 all        170       1004      0.817      0.625      0.696      0.287\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    24/149     1.69G   0.04384   0.03082  0.007855         1       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.39it/s]\n",
            "                 all        170       1004       0.84      0.619       0.68      0.245\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    25/149     1.69G   0.04408   0.03137  0.007674        23       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.34it/s]\n",
            "                 all        170       1004       0.88      0.662      0.735       0.32\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    26/149     1.69G   0.04376   0.03093  0.007991        10       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.41it/s]\n",
            "                 all        170       1004      0.769      0.641      0.671      0.255\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    27/149     1.69G   0.04461   0.03293  0.007568        44       416: 100% 128/128 [01:12<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.36it/s]\n",
            "                 all        170       1004       0.87      0.638      0.729      0.324\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    28/149     1.69G   0.04414   0.03093  0.007756         3       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.44it/s]\n",
            "                 all        170       1004      0.852      0.649      0.728      0.315\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    29/149     1.69G   0.04288   0.03207  0.007473         8       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.30it/s]\n",
            "                 all        170       1004      0.872      0.646      0.722      0.302\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    30/149     1.69G   0.04362   0.03165  0.008004         7       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.28it/s]\n",
            "                 all        170       1004      0.848      0.609      0.669        0.2\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    31/149     1.69G   0.04359   0.03119   0.00767        19       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.34it/s]\n",
            "                 all        170       1004      0.777      0.673      0.713      0.318\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    32/149     1.69G   0.04379   0.03156  0.007285        11       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.41it/s]\n",
            "                 all        170       1004      0.875       0.65      0.719      0.277\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    33/149     1.69G   0.04353   0.02988  0.007174        13       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.42it/s]\n",
            "                 all        170       1004       0.86      0.644       0.73      0.322\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    34/149     1.69G   0.04264   0.03114  0.007594         9       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.35it/s]\n",
            "                 all        170       1004      0.814      0.656      0.715      0.327\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    35/149     1.69G   0.04286   0.03033  0.007179         7       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.39it/s]\n",
            "                 all        170       1004      0.812      0.717      0.765      0.363\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    36/149     1.69G   0.04321   0.03091  0.007568        10       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.38it/s]\n",
            "                 all        170       1004      0.866      0.666       0.74      0.306\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    37/149     1.69G    0.0419   0.03072  0.006968        17       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.41it/s]\n",
            "                 all        170       1004      0.794      0.702      0.755      0.322\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    38/149     1.69G   0.04238   0.03075  0.007304        10       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.43it/s]\n",
            "                 all        170       1004      0.877      0.662       0.74       0.31\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    39/149     1.69G    0.0422   0.02997  0.006842         3       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.37it/s]\n",
            "                 all        170       1004      0.821      0.684      0.744      0.331\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    40/149     1.69G   0.04182   0.03094  0.006975        10       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.41it/s]\n",
            "                 all        170       1004      0.835       0.66      0.726       0.32\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    41/149     1.69G   0.04243   0.03148  0.007315        14       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.29it/s]\n",
            "                 all        170       1004      0.825      0.694      0.746      0.323\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    42/149     1.69G   0.04193   0.03028  0.007496         8       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.37it/s]\n",
            "                 all        170       1004      0.838      0.682       0.74      0.288\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    43/149     1.69G   0.04163   0.03017  0.006961        11       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.38it/s]\n",
            "                 all        170       1004      0.877      0.666      0.746       0.33\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    44/149     1.69G   0.04153   0.02976  0.006867         6       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.38it/s]\n",
            "                 all        170       1004      0.839      0.696      0.756      0.343\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    45/149     1.69G   0.04159   0.02989  0.006573        19       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.28it/s]\n",
            "                 all        170       1004      0.811      0.677      0.717      0.315\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    46/149     1.69G   0.04215   0.03092  0.007041        24       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.39it/s]\n",
            "                 all        170       1004      0.835      0.681      0.744       0.34\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    47/149     1.69G   0.04057   0.02989  0.006955        18       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.36it/s]\n",
            "                 all        170       1004      0.826        0.7      0.755      0.339\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    48/149     1.69G   0.04043   0.03014  0.006845        14       416: 100% 128/128 [01:12<00:00,  1.76it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.30it/s]\n",
            "                 all        170       1004      0.855      0.685      0.749       0.31\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    49/149     1.69G   0.04199   0.03078  0.006384        16       416: 100% 128/128 [01:12<00:00,  1.76it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.37it/s]\n",
            "                 all        170       1004      0.785      0.746      0.771      0.391\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    50/149     1.69G    0.0404   0.02994  0.006534        18       416: 100% 128/128 [01:12<00:00,  1.76it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.33it/s]\n",
            "                 all        170       1004       0.86      0.681      0.756      0.325\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    51/149     1.69G   0.04065   0.03058  0.006354        16       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.34it/s]\n",
            "                 all        170       1004      0.865       0.69      0.763      0.326\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    52/149     1.69G   0.04088   0.02915  0.006616         6       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.42it/s]\n",
            "                 all        170       1004      0.846      0.662      0.732      0.324\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    53/149     1.69G    0.0406   0.02929  0.006659        11       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.38it/s]\n",
            "                 all        170       1004       0.85      0.697      0.764       0.33\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    54/149     1.69G   0.04054   0.02971  0.006845        20       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.42it/s]\n",
            "                 all        170       1004      0.851      0.656      0.728      0.286\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    55/149     1.69G   0.04075   0.03034  0.006677        16       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.32it/s]\n",
            "                 all        170       1004       0.85      0.705      0.756      0.371\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    56/149     1.69G   0.04135   0.03004  0.006767        11       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.44it/s]\n",
            "                 all        170       1004      0.858      0.663      0.737      0.314\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    57/149     1.69G   0.04026   0.02946  0.006197        25       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.37it/s]\n",
            "                 all        170       1004       0.88      0.692       0.76      0.332\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    58/149     1.69G   0.04005   0.03023  0.005879        13       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.38it/s]\n",
            "                 all        170       1004      0.912      0.726      0.793      0.372\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    59/149     1.69G   0.04005   0.02946  0.005951         3       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.40it/s]\n",
            "                 all        170       1004      0.841      0.714      0.769      0.334\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    60/149     1.69G   0.04066   0.02928  0.006289        11       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.36it/s]\n",
            "                 all        170       1004       0.84      0.738      0.779      0.388\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    61/149     1.69G   0.03939   0.02925   0.00595        12       416: 100% 128/128 [01:12<00:00,  1.76it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.35it/s]\n",
            "                 all        170       1004      0.835      0.724      0.775      0.356\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    62/149     1.69G   0.03981    0.0296  0.006262        21       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.44it/s]\n",
            "                 all        170       1004       0.86      0.703      0.774      0.365\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    63/149     1.69G   0.03944   0.02808  0.006258        10       416: 100% 128/128 [01:12<00:00,  1.76it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.33it/s]\n",
            "                 all        170       1004      0.848      0.698       0.77      0.365\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    64/149     1.69G   0.03922    0.0293  0.005963         9       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.39it/s]\n",
            "                 all        170       1004      0.903      0.684      0.764      0.327\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    65/149     1.69G   0.03932   0.02933  0.005897         8       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.40it/s]\n",
            "                 all        170       1004      0.851      0.726      0.778      0.343\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    66/149     1.69G   0.03878   0.02885  0.005849        29       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.39it/s]\n",
            "                 all        170       1004       0.83      0.695      0.757      0.332\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    67/149     1.69G   0.03982   0.02955  0.006275         3       416: 100% 128/128 [01:12<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.43it/s]\n",
            "                 all        170       1004      0.841      0.715      0.766      0.356\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    68/149     1.69G   0.03921    0.0289  0.005816         1       416: 100% 128/128 [01:12<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.44it/s]\n",
            "                 all        170       1004      0.865      0.697      0.776       0.39\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    69/149     1.69G   0.03919   0.02853  0.005701        26       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.34it/s]\n",
            "                 all        170       1004      0.885      0.724      0.802      0.376\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    70/149     1.69G   0.03882   0.02977  0.005999         7       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.37it/s]\n",
            "                 all        170       1004      0.876      0.715      0.787      0.364\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    71/149     1.69G   0.03905   0.02905  0.005907        10       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.39it/s]\n",
            "                 all        170       1004      0.872      0.701      0.792      0.376\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    72/149     1.69G    0.0392   0.02897   0.00574         6       416: 100% 128/128 [01:12<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.35it/s]\n",
            "                 all        170       1004      0.899      0.724      0.805      0.342\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    73/149     1.69G   0.03889   0.02821  0.005539         6       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.54it/s]\n",
            "                 all        170       1004       0.84      0.709      0.784      0.363\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    74/149     1.69G   0.03886   0.02854  0.005622         3       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.46it/s]\n",
            "                 all        170       1004      0.871      0.716      0.788      0.375\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    75/149     1.69G   0.03828   0.02858  0.005396         4       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.44it/s]\n",
            "                 all        170       1004      0.879      0.716      0.792       0.36\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    76/149     1.69G   0.03813   0.02832  0.005436        16       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.42it/s]\n",
            "                 all        170       1004      0.897      0.708      0.781      0.349\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    77/149     1.69G   0.03737   0.02825   0.00542        17       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.52it/s]\n",
            "                 all        170       1004      0.916      0.709      0.804      0.387\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    78/149     1.69G   0.03836   0.02849  0.005796        13       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.53it/s]\n",
            "                 all        170       1004      0.903       0.69      0.789      0.371\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    79/149     1.69G   0.03752   0.02837  0.005222        28       416: 100% 128/128 [01:10<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.50it/s]\n",
            "                 all        170       1004      0.843      0.744      0.803       0.37\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    80/149     1.69G   0.03854   0.02768  0.005487        11       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.45it/s]\n",
            "                 all        170       1004      0.875      0.717      0.793      0.399\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    81/149     1.69G   0.03776   0.02768  0.005147        26       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.44it/s]\n",
            "                 all        170       1004      0.873      0.746      0.811      0.416\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    82/149     1.69G   0.03797   0.02823  0.005291        11       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.50it/s]\n",
            "                 all        170       1004      0.877      0.709      0.791       0.38\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    83/149     1.69G   0.03787   0.02736  0.005205         6       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.42it/s]\n",
            "                 all        170       1004      0.874      0.732      0.794      0.377\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    84/149     1.69G   0.03785   0.02811   0.00509         6       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.42it/s]\n",
            "                 all        170       1004      0.885       0.73      0.806      0.376\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    85/149     1.69G    0.0371   0.02827  0.005038         8       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.45it/s]\n",
            "                 all        170       1004      0.879      0.734      0.802      0.392\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    86/149     1.69G   0.03711   0.02837  0.005315        24       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.43it/s]\n",
            "                 all        170       1004      0.816      0.759      0.812      0.392\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    87/149     1.69G   0.03742   0.02792  0.005277        14       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.47it/s]\n",
            "                 all        170       1004      0.848      0.736      0.791      0.363\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    88/149     1.69G   0.03694   0.02754  0.005232         5       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.48it/s]\n",
            "                 all        170       1004      0.904      0.729      0.811      0.385\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    89/149     1.69G    0.0363   0.02715  0.005071        12       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.50it/s]\n",
            "                 all        170       1004      0.866      0.741      0.816      0.399\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    90/149     1.69G   0.03687   0.02694  0.005134        10       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.45it/s]\n",
            "                 all        170       1004      0.862      0.761      0.811      0.406\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    91/149     1.69G   0.03635   0.02748  0.004971        18       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.40it/s]\n",
            "                 all        170       1004      0.872      0.738      0.808      0.389\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    92/149     1.69G   0.03635    0.0268  0.004584         5       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.49it/s]\n",
            "                 all        170       1004      0.906      0.708        0.8      0.356\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    93/149     1.69G   0.03624   0.02826  0.004829        24       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.47it/s]\n",
            "                 all        170       1004      0.852      0.758      0.812      0.404\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    94/149     1.69G    0.0361   0.02712  0.004914         8       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.51it/s]\n",
            "                 all        170       1004      0.852      0.753      0.806      0.386\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    95/149     1.69G   0.03634   0.02748  0.005059         7       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.50it/s]\n",
            "                 all        170       1004      0.869      0.741      0.806      0.377\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    96/149     1.69G   0.03638   0.02697  0.004749        15       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.51it/s]\n",
            "                 all        170       1004      0.879      0.735      0.813        0.4\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    97/149     1.69G   0.03566     0.027  0.004796         3       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.50it/s]\n",
            "                 all        170       1004      0.878      0.753      0.815      0.407\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    98/149     1.69G   0.03523    0.0262  0.004358         8       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.46it/s]\n",
            "                 all        170       1004      0.878      0.752      0.813       0.39\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "    99/149     1.69G   0.03533    0.0259   0.00469        23       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.56it/s]\n",
            "                 all        170       1004       0.89      0.728      0.806      0.398\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   100/149     1.69G   0.03568   0.02686  0.004638         7       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.45it/s]\n",
            "                 all        170       1004      0.852      0.751      0.804      0.412\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   101/149     1.69G    0.0356   0.02646  0.004612        12       416: 100% 128/128 [01:12<00:00,  1.77it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.47it/s]\n",
            "                 all        170       1004      0.908      0.736      0.818      0.394\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   102/149     1.69G     0.035   0.02646  0.004707        10       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.47it/s]\n",
            "                 all        170       1004      0.877      0.736      0.814      0.351\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   103/149     1.69G   0.03532   0.02616   0.00474        10       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.49it/s]\n",
            "                 all        170       1004      0.877      0.778      0.833       0.39\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   104/149     1.69G   0.03409   0.02495  0.004286         6       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.51it/s]\n",
            "                 all        170       1004      0.939       0.72      0.822      0.391\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   105/149     1.69G   0.03519   0.02604  0.004356         9       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.52it/s]\n",
            "                 all        170       1004      0.893      0.752      0.832      0.415\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   106/149     1.69G   0.03463   0.02651  0.004439        29       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.48it/s]\n",
            "                 all        170       1004      0.861      0.778      0.819      0.404\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   107/149     1.69G   0.03501   0.02677  0.004341         6       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.49it/s]\n",
            "                 all        170       1004       0.84      0.792      0.828      0.413\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   108/149     1.69G   0.03386   0.02608  0.004228        24       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.52it/s]\n",
            "                 all        170       1004      0.863      0.768      0.817      0.388\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   109/149     1.69G   0.03455   0.02591  0.004286         5       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.53it/s]\n",
            "                 all        170       1004      0.921      0.743      0.826      0.379\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   110/149     1.69G   0.03429   0.02574  0.004103        12       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.51it/s]\n",
            "                 all        170       1004      0.888      0.767      0.824      0.406\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   111/149     1.69G   0.03386   0.02507  0.004349         8       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.51it/s]\n",
            "                 all        170       1004      0.855      0.793      0.836      0.399\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   112/149     1.69G    0.0335   0.02652  0.004129         4       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.48it/s]\n",
            "                 all        170       1004      0.877      0.781      0.837      0.405\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   113/149     1.69G   0.03411   0.02551  0.003974        12       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.50it/s]\n",
            "                 all        170       1004      0.865      0.796      0.839      0.403\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   114/149     1.69G    0.0337   0.02512  0.003728         3       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.43it/s]\n",
            "                 all        170       1004       0.88      0.765      0.837      0.416\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   115/149     1.69G   0.03332   0.02625  0.004037         7       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.52it/s]\n",
            "                 all        170       1004       0.87      0.775      0.824      0.403\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   116/149     1.69G   0.03354   0.02475  0.004302        10       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.51it/s]\n",
            "                 all        170       1004      0.902      0.745      0.827      0.394\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   117/149     1.69G   0.03368   0.02557  0.004213        13       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.41it/s]\n",
            "                 all        170       1004      0.881      0.765       0.83      0.392\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   118/149     1.69G   0.03282   0.02533  0.003976         9       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.47it/s]\n",
            "                 all        170       1004      0.913      0.763      0.833      0.411\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   119/149     1.69G   0.03262   0.02404  0.003344         5       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.52it/s]\n",
            "                 all        170       1004      0.906       0.77      0.832      0.392\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   120/149     1.69G   0.03245   0.02512  0.003705         9       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.56it/s]\n",
            "                 all        170       1004      0.872      0.785      0.837      0.407\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   121/149     1.69G   0.03267   0.02451  0.003996         7       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.46it/s]\n",
            "                 all        170       1004      0.888      0.779      0.839       0.41\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   122/149     1.69G   0.03306   0.02497   0.00377        18       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.48it/s]\n",
            "                 all        170       1004      0.892       0.78      0.841      0.418\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   123/149     1.69G   0.03321   0.02569  0.003833         9       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.49it/s]\n",
            "                 all        170       1004      0.876      0.794      0.848      0.426\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   124/149     1.69G   0.03261   0.02472  0.003783        18       416: 100% 128/128 [01:11<00:00,  1.78it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.52it/s]\n",
            "                 all        170       1004      0.893      0.766      0.834      0.411\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   125/149     1.69G   0.03232   0.02528  0.003538         4       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.48it/s]\n",
            "                 all        170       1004      0.865      0.795      0.843      0.419\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   126/149     1.69G   0.03216   0.02441  0.003474         7       416: 100% 128/128 [01:10<00:00,  1.81it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.51it/s]\n",
            "                 all        170       1004      0.898      0.759       0.84      0.431\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   127/149     1.69G   0.03233   0.02404  0.003568        11       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.39it/s]\n",
            "                 all        170       1004      0.887      0.774      0.849      0.416\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   128/149     1.69G   0.03214   0.02424  0.003461         5       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.49it/s]\n",
            "                 all        170       1004      0.868      0.804      0.843      0.409\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   129/149     1.69G   0.03206   0.02421  0.003638        13       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.48it/s]\n",
            "                 all        170       1004      0.889      0.787      0.848      0.413\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   130/149     1.69G   0.03257   0.02457  0.003776         5       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.44it/s]\n",
            "                 all        170       1004      0.878        0.8      0.845      0.436\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   131/149     1.69G   0.03173   0.02516  0.003404         8       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.46it/s]\n",
            "                 all        170       1004      0.893      0.772      0.843      0.422\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   132/149     1.69G   0.03184   0.02467  0.003352        18       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.47it/s]\n",
            "                 all        170       1004      0.865      0.807      0.847      0.434\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   133/149     1.69G   0.03125   0.02306   0.00351        14       416: 100% 128/128 [01:10<00:00,  1.81it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.49it/s]\n",
            "                 all        170       1004       0.85      0.802       0.84      0.415\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   134/149     1.69G    0.0317   0.02416  0.003628         9       416: 100% 128/128 [01:10<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.48it/s]\n",
            "                 all        170       1004      0.858      0.796      0.842      0.419\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   135/149     1.69G    0.0316   0.02384  0.003233        15       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.48it/s]\n",
            "                 all        170       1004      0.876      0.786      0.837      0.409\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   136/149     1.69G   0.03123   0.02355  0.003422        14       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.45it/s]\n",
            "                 all        170       1004        0.9      0.777      0.841      0.426\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   137/149     1.69G   0.03121   0.02452  0.003297        23       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.50it/s]\n",
            "                 all        170       1004      0.879      0.793       0.85      0.417\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   138/149     1.69G   0.03082    0.0239  0.003196         5       416: 100% 128/128 [01:11<00:00,  1.79it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.50it/s]\n",
            "                 all        170       1004      0.895      0.773      0.851      0.428\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   139/149     1.69G   0.03042   0.02298  0.003105         8       416: 100% 128/128 [01:10<00:00,  1.81it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.53it/s]\n",
            "                 all        170       1004      0.878      0.794      0.851      0.453\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   140/149     1.69G   0.03069   0.02388  0.003092        19       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.49it/s]\n",
            "                 all        170       1004      0.906      0.765      0.847      0.422\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   141/149     1.69G   0.03137   0.02409  0.003342         6       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.47it/s]\n",
            "                 all        170       1004       0.88      0.801      0.845      0.419\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   142/149     1.69G   0.03059   0.02281  0.003088         5       416: 100% 128/128 [01:10<00:00,  1.81it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.47it/s]\n",
            "                 all        170       1004      0.872      0.784      0.831      0.417\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   143/149     1.69G   0.03064   0.02384  0.003187        34       416: 100% 128/128 [01:10<00:00,  1.81it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.48it/s]\n",
            "                 all        170       1004      0.878      0.774      0.837      0.428\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   144/149     1.69G   0.03071   0.02364  0.003149        23       416: 100% 128/128 [01:10<00:00,  1.81it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.58it/s]\n",
            "                 all        170       1004      0.867      0.792      0.839      0.422\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   145/149     1.69G   0.03046   0.02345  0.003045        17       416: 100% 128/128 [01:10<00:00,  1.81it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.50it/s]\n",
            "                 all        170       1004      0.852      0.804      0.844      0.444\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   146/149     1.69G    0.0305    0.0242  0.003257        23       416: 100% 128/128 [01:10<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.52it/s]\n",
            "                 all        170       1004      0.848      0.792       0.84      0.435\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   147/149     1.69G   0.03019   0.02393   0.00322         9       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.33it/s]\n",
            "                 all        170       1004      0.872      0.787      0.841      0.433\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   148/149     1.69G   0.03032   0.02373  0.003047        21       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.46it/s]\n",
            "                 all        170       1004      0.848      0.793      0.834      0.439\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "   149/149     1.69G    0.0304   0.02335  0.003239        13       416: 100% 128/128 [01:11<00:00,  1.80it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:02<00:00,  2.49it/s]\n",
            "                 all        170       1004      0.895      0.779      0.843      0.434\n",
            "\n",
            "150 epochs completed in 3.124 hours.\n",
            "Optimizer stripped from yolov5/runs/train/robo4_epoch150_s/weights/last.pt, 14.3MB\n",
            "Optimizer stripped from yolov5/runs/train/robo4_epoch150_s/weights/best.pt, 14.3MB\n",
            "\n",
            "Validating yolov5/runs/train/robo4_epoch150_s/weights/best.pt...\n",
            "Fusing layers... \n",
            "Model Summary: 213 layers, 7015519 parameters, 0 gradients, 15.8 GFLOPs\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 6/6 [00:08<00:00,  1.46s/it]\n",
            "                 all        170       1004      0.877      0.793       0.85      0.453\n",
            "                 bad        170        217      0.834      0.742      0.786      0.401\n",
            "                good        170        787       0.92      0.845      0.914      0.506\n",
            "Results saved to \u001b[1myolov5/runs/train/robo4_epoch150_s\u001b[0m\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "4dn4R0urR2j8",
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "452feb48-89a2-4fcb-b5da-12e11e059e8f"
      },
      "source": [
        "!python yolov5/detect.py --weights yolov5/runs/train/robo4_epoch150_s/weights/best.pt --source 0 --iou-thres 0.3 --conf-thres 0.6"
      ],
      "execution_count": 12,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "\u001b[34m\u001b[1mdetect: \u001b[0mweights=['yolov5/runs/train/robo4_epoch150_s/weights/best.pt'], source=0, imgsz=[640, 640], conf_thres=0.6, iou_thres=0.3, max_det=1000, device=, view_img=False, save_txt=False, save_conf=False, save_crop=False, nosave=False, classes=None, agnostic_nms=False, augment=False, visualize=False, update=False, project=yolov5/runs/detect, name=exp, exist_ok=False, line_thickness=3, hide_labels=False, hide_conf=False, half=False, dnn=False\n",
            "YOLOv5 🚀 v6.0-113-g5ca5dd4 torch 1.10.0+cu111 CUDA:0 (Tesla K80, 11441MiB)\n",
            "\n",
            "Fusing layers... \n",
            "Model Summary: 213 layers, 7015519 parameters, 0 gradients, 15.8 GFLOPs\n",
            "WARNING: Environment does not support cv2.imshow() or PIL Image.show() image displays\n",
            "cv2.imshow() is disabled in Google Colab environments\n",
            "[ WARN:0] global /tmp/pip-req-build-3129w7z7/opencv/modules/videoio/src/cap_v4l.cpp (890) open VIDEOIO(V4L2:/dev/video0): can't open camera by index\n",
            "Traceback (most recent call last):\n",
            "  File \"yolov5/detect.py\", line 243, in <module>\n",
            "    main(opt)\n",
            "  File \"yolov5/detect.py\", line 238, in main\n",
            "    run(**vars(opt))\n",
            "  File \"/usr/local/lib/python3.7/dist-packages/torch/autograd/grad_mode.py\", line 28, in decorate_context\n",
            "    return func(*args, **kwargs)\n",
            "  File \"yolov5/detect.py\", line 92, in run\n",
            "    dataset = LoadStreams(source, img_size=imgsz, stride=stride, auto=pt and not jit)\n",
            "  File \"/content/yolov5/utils/datasets.py\", line 309, in __init__\n",
            "    assert cap.isOpened(), f'{st}Failed to open {s}'\n",
            "AssertionError: 1/1: 0... Failed to open 0\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "5SqNUkHNMlsC"
      },
      "source": [
        ""
      ],
      "execution_count": null,
      "outputs": []
    }
  ]
}