Подготовка_данных.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "provenance": [],
      "gpuType": "T4"
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    },
    "accelerator": "GPU"
  },
  "cells": [
    {
      "cell_type": "markdown",
      "source": [
        "# Подготовка датафрейма с разметкой изображений на основании xml. файла из программы CVAT"
      ],
      "metadata": {
        "id": "Y3gqP08yJyC8"
      }
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "ojixP-LdNvpw"
      },
      "outputs": [],
      "source": [
        "import os\n",
        "import glob\n",
        "import imageio\n",
        "import shutil\n",
        "import xml.etree.ElementTree as ET\n",
        "import pandas as pd\n",
        "import matplotlib.pyplot as plt\n",
        "from PIL import Image\n"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "img_path = '/content/images'\n",
        "annotations = '/content/annotation'"
      ],
      "metadata": {
        "id": "x-SqbMnQOgfk"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "# Загружаем изображения как массивы NumPy и добавляем их в список images\n",
        "images = []\n",
        "for file in glob.glob(os.path.join(img_path, '*.jpg')):\n",
        "    images.append(imageio.imread(file))"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "YTREMFXxQm3L",
        "outputId": "76b1323c-18ff-415f-947b-d5b4fcde75c4"
      },
      "execution_count": null,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "<ipython-input-12-2ce9de1eb72a>:4: DeprecationWarning: Starting with ImageIO v3 the behavior of this function will switch to that of iio.v3.imread. To keep the current behavior (and make this warning disappear) use `import imageio.v2 as imageio` or call `imageio.v2.imread` directly.\n",
            "  images.append(imageio.imread(file))\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "# Сколько у нас изображений\n",
        "print('We have {} images'.format(len(images)))\n"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "VzpO2qypQp_n",
        "outputId": "2224810e-cff5-4918-d594-2b7ae094cded"
      },
      "execution_count": null,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "We have 210 images\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "# Функция для конвертации XML в DataFrame\n",
        "def xml_to_csv(xml_file):\n",
        "    tree = ET.parse(xml_file)\n",
        "    root = tree.getroot()\n",
        "\n",
        "    filename = root.find('filename').text\n",
        "    width = int(root.find('imagesize').find('ncols').text)\n",
        "    height = int(root.find('imagesize').find('nrows').text)\n",
        "\n",
        "    data = []\n",
        "    for obj in root.findall('object'):\n",
        "        obj_name = obj.find('name').text\n",
        "        xmin = ymin = xmax = ymax = None\n",
        "\n",
        "        if len(obj.findall('polygon')) > 0:\n",
        "            polygon = obj.find('polygon')\n",
        "            pts = polygon.findall('pt')\n",
        "            x_coords = [float(pt.find('x').text) for pt in pts]\n",
        "            y_coords = [float(pt.find('y').text) for pt in pts]\n",
        "            xmin = min(x_coords)\n",
        "            xmax = max(x_coords)\n",
        "            ymin = min(y_coords)\n",
        "            ymax = max(y_coords)\n",
        "\n",
        "        data.append([filename, width, height, obj_name, xmin, ymin, xmax, ymax])\n",
        "\n",
        "    column_names = ['filename', 'width', 'height', 'class', 'xmin', 'ymin', 'xmax', 'ymax']\n",
        "    df = pd.DataFrame(data, columns=column_names)\n",
        "    return df"
      ],
      "metadata": {
        "id": "b5w1SidaRCss"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "# Создаем пустой DataFrame для сохранения данных\n",
        "labels_df = pd.DataFrame()"
      ],
      "metadata": {
        "id": "yn_ZmkeRSAuI"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "# Пройдем по всем XML файлам в папке с аннотациями и конвертируем их в CSV\n",
        "for file in os.listdir(annotations):\n",
        "    if file.endswith('.xml'):\n",
        "        df = xml_to_csv(os.path.join(annotations, file))\n",
        "        labels_df = pd.concat([labels_df, df], ignore_index=True)\n",
        "\n",
        "\n",
        "print('Successfully converted xml to csv.')\n",
        "print(labels_df)"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "5jWOUDiaRMg2",
        "outputId": "227db8d9-0f0e-469c-fe86-f0d9f082fd33"
      },
      "execution_count": null,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Successfully converted xml to csv.\n",
            "        filename  width  height  class    xmin    ymin    xmax    ymax\n",
            "0    2 (112).jpg    854     480  break  505.52  177.12  537.08  233.02\n",
            "1    2 (112).jpg    854     480  crack  126.72   39.05  548.75  456.60\n",
            "2    2 (112).jpg    854     480  joint  507.45    1.20  722.75  479.80\n",
            "3    2 (112).jpg    854     480  crack  632.60    2.80  725.70  352.80\n",
            "4     3 (86).jpg    854     480  crack  267.44   34.29  588.16  284.40\n",
            "..           ...    ...     ...    ...     ...     ...     ...     ...\n",
            "919  3 (127).jpg    854     480  break  332.40  406.60  412.85  423.80\n",
            "920  2 (114).jpg    854     480  break  507.34  174.97  538.88  224.42\n",
            "921  2 (114).jpg    854     480  crack  126.72   37.20  555.60  456.60\n",
            "922  2 (114).jpg    854     480  joint  513.20    1.90  728.10  480.00\n",
            "923  2 (114).jpg    854     480  crack  661.88    2.67  726.30  296.70\n",
            "\n",
            "[924 rows x 8 columns]\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "# Загрузите DataFrame с аннотациями\n",
        "labels_df"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 424
        },
        "id": "nAFLsaFZB8Jg",
        "outputId": "d9651af8-96cb-4a74-87b5-0c3ee43d41ff"
      },
      "execution_count": null,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "        filename  width  height  class    xmin    ymin    xmax    ymax\n",
              "0    2 (112).jpg    854     480  break  505.52  177.12  537.08  233.02\n",
              "1    2 (112).jpg    854     480  crack  126.72   39.05  548.75  456.60\n",
              "2    2 (112).jpg    854     480  joint  507.45    1.20  722.75  479.80\n",
              "3    2 (112).jpg    854     480  crack  632.60    2.80  725.70  352.80\n",
              "4     3 (86).jpg    854     480  crack  267.44   34.29  588.16  284.40\n",
              "..           ...    ...     ...    ...     ...     ...     ...     ...\n",
              "919  3 (127).jpg    854     480  break  332.40  406.60  412.85  423.80\n",
              "920  2 (114).jpg    854     480  break  507.34  174.97  538.88  224.42\n",
              "921  2 (114).jpg    854     480  crack  126.72   37.20  555.60  456.60\n",
              "922  2 (114).jpg    854     480  joint  513.20    1.90  728.10  480.00\n",
              "923  2 (114).jpg    854     480  crack  661.88    2.67  726.30  296.70\n",
              "\n",
              "[924 rows x 8 columns]"
            ],
            "text/html": [
              "\n",
              "  <div id=\"df-26b7391a-30ef-492e-b943-d9390b67eb70\" class=\"colab-df-container\">\n",
              "    <div>\n",
              "<style scoped>\n",
              "    .dataframe tbody tr th:only-of-type {\n",
              "        vertical-align: middle;\n",
              "    }\n",
              "\n",
              "    .dataframe tbody tr th {\n",
              "        vertical-align: top;\n",
              "    }\n",
              "\n",
              "    .dataframe thead th {\n",
              "        text-align: right;\n",
              "    }\n",
              "</style>\n",
              "<table border=\"1\" class=\"dataframe\">\n",
              "  <thead>\n",
              "    <tr style=\"text-align: right;\">\n",
              "      <th></th>\n",
              "      <th>filename</th>\n",
              "      <th>width</th>\n",
              "      <th>height</th>\n",
              "      <th>class</th>\n",
              "      <th>xmin</th>\n",
              "      <th>ymin</th>\n",
              "      <th>xmax</th>\n",
              "      <th>ymax</th>\n",
              "    </tr>\n",
              "  </thead>\n",
              "  <tbody>\n",
              "    <tr>\n",
              "      <th>0</th>\n",
              "      <td>2 (112).jpg</td>\n",
              "      <td>854</td>\n",
              "      <td>480</td>\n",
              "      <td>break</td>\n",
              "      <td>505.52</td>\n",
              "      <td>177.12</td>\n",
              "      <td>537.08</td>\n",
              "      <td>233.02</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>1</th>\n",
              "      <td>2 (112).jpg</td>\n",
              "      <td>854</td>\n",
              "      <td>480</td>\n",
              "      <td>crack</td>\n",
              "      <td>126.72</td>\n",
              "      <td>39.05</td>\n",
              "      <td>548.75</td>\n",
              "      <td>456.60</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>2</th>\n",
              "      <td>2 (112).jpg</td>\n",
              "      <td>854</td>\n",
              "      <td>480</td>\n",
              "      <td>joint</td>\n",
              "      <td>507.45</td>\n",
              "      <td>1.20</td>\n",
              "      <td>722.75</td>\n",
              "      <td>479.80</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>3</th>\n",
              "      <td>2 (112).jpg</td>\n",
              "      <td>854</td>\n",
              "      <td>480</td>\n",
              "      <td>crack</td>\n",
              "      <td>632.60</td>\n",
              "      <td>2.80</td>\n",
              "      <td>725.70</td>\n",
              "      <td>352.80</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>4</th>\n",
              "      <td>3 (86).jpg</td>\n",
              "      <td>854</td>\n",
              "      <td>480</td>\n",
              "      <td>crack</td>\n",
              "      <td>267.44</td>\n",
              "      <td>34.29</td>\n",
              "      <td>588.16</td>\n",
              "      <td>284.40</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>...</th>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>919</th>\n",
              "      <td>3 (127).jpg</td>\n",
              "      <td>854</td>\n",
              "      <td>480</td>\n",
              "      <td>break</td>\n",
              "      <td>332.40</td>\n",
              "      <td>406.60</td>\n",
              "      <td>412.85</td>\n",
              "      <td>423.80</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>920</th>\n",
              "      <td>2 (114).jpg</td>\n",
              "      <td>854</td>\n",
              "      <td>480</td>\n",
              "      <td>break</td>\n",
              "      <td>507.34</td>\n",
              "      <td>174.97</td>\n",
              "      <td>538.88</td>\n",
              "      <td>224.42</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>921</th>\n",
              "      <td>2 (114).jpg</td>\n",
              "      <td>854</td>\n",
              "      <td>480</td>\n",
              "      <td>crack</td>\n",
              "      <td>126.72</td>\n",
              "      <td>37.20</td>\n",
              "      <td>555.60</td>\n",
              "      <td>456.60</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>922</th>\n",
              "      <td>2 (114).jpg</td>\n",
              "      <td>854</td>\n",
              "      <td>480</td>\n",
              "      <td>joint</td>\n",
              "      <td>513.20</td>\n",
              "      <td>1.90</td>\n",
              "      <td>728.10</td>\n",
              "      <td>480.00</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>923</th>\n",
              "      <td>2 (114).jpg</td>\n",
              "      <td>854</td>\n",
              "      <td>480</td>\n",
              "      <td>crack</td>\n",
              "      <td>661.88</td>\n",
              "      <td>2.67</td>\n",
              "      <td>726.30</td>\n",
              "      <td>296.70</td>\n",
              "    </tr>\n",
              "  </tbody>\n",
              "</table>\n",
              "<p>924 rows × 8 columns</p>\n",
              "</div>\n",
              "    <div class=\"colab-df-buttons\">\n",
              "\n",
              "  <div class=\"colab-df-container\">\n",
              "    <button class=\"colab-df-convert\" onclick=\"convertToInteractive('df-26b7391a-30ef-492e-b943-d9390b67eb70')\"\n",
              "            title=\"Convert this dataframe to an interactive table.\"\n",
              "            style=\"display:none;\">\n",
              "\n",
              "  <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\" viewBox=\"0 -960 960 960\">\n",
              "    <path d=\"M120-120v-720h720v720H120Zm60-500h600v-160H180v160Zm220 220h160v-160H400v160Zm0 220h160v-160H400v160ZM180-400h160v-160H180v160Zm440 0h160v-160H620v160ZM180-180h160v-160H180v160Zm440 0h160v-160H620v160Z\"/>\n",
              "  </svg>\n",
              "    </button>\n",
              "\n",
              "  <style>\n",
              "    .colab-df-container {\n",
              "      display:flex;\n",
              "      gap: 12px;\n",
              "    }\n",
              "\n",
              "    .colab-df-convert {\n",
              "      background-color: #E8F0FE;\n",
              "      border: none;\n",
              "      border-radius: 50%;\n",
              "      cursor: pointer;\n",
              "      display: none;\n",
              "      fill: #1967D2;\n",
              "      height: 32px;\n",
              "      padding: 0 0 0 0;\n",
              "      width: 32px;\n",
              "    }\n",
              "\n",
              "    .colab-df-convert:hover {\n",
              "      background-color: #E2EBFA;\n",
              "      box-shadow: 0px 1px 2px rgba(60, 64, 67, 0.3), 0px 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
              "      fill: #174EA6;\n",
              "    }\n",
              "\n",
              "    .colab-df-buttons div {\n",
              "      margin-bottom: 4px;\n",
              "    }\n",
              "\n",
              "    [theme=dark] .colab-df-convert {\n",
              "      background-color: #3B4455;\n",
              "      fill: #D2E3FC;\n",
              "    }\n",
              "\n",
              "    [theme=dark] .colab-df-convert:hover {\n",
              "      background-color: #434B5C;\n",
              "      box-shadow: 0px 1px 3px 1px rgba(0, 0, 0, 0.15);\n",
              "      filter: drop-shadow(0px 1px 2px rgba(0, 0, 0, 0.3));\n",
              "      fill: #FFFFFF;\n",
              "    }\n",
              "  </style>\n",
              "\n",
              "    <script>\n",
              "      const buttonEl =\n",
              "        document.querySelector('#df-26b7391a-30ef-492e-b943-d9390b67eb70 button.colab-df-convert');\n",
              "      buttonEl.style.display =\n",
              "        google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
              "\n",
              "      async function convertToInteractive(key) {\n",
              "        const element = document.querySelector('#df-26b7391a-30ef-492e-b943-d9390b67eb70');\n",
              "        const dataTable =\n",
              "          await google.colab.kernel.invokeFunction('convertToInteractive',\n",
              "                                                    [key], {});\n",
              "        if (!dataTable) return;\n",
              "\n",
              "        const docLinkHtml = 'Like what you see? Visit the ' +\n",
              "          '<a target=\"_blank\" href=https://colab.research.google.com/notebooks/data_table.ipynb>data table notebook</a>'\n",
              "          + ' to learn more about interactive tables.';\n",
              "        element.innerHTML = '';\n",
              "        dataTable['output_type'] = 'display_data';\n",
              "        await google.colab.output.renderOutput(dataTable, element);\n",
              "        const docLink = document.createElement('div');\n",
              "        docLink.innerHTML = docLinkHtml;\n",
              "        element.appendChild(docLink);\n",
              "      }\n",
              "    </script>\n",
              "  </div>\n",
              "\n",
              "\n",
              "<div id=\"df-46a6bdac-490e-4fb6-99be-12127504cd70\">\n",
              "  <button class=\"colab-df-quickchart\" onclick=\"quickchart('df-46a6bdac-490e-4fb6-99be-12127504cd70')\"\n",
              "            title=\"Suggest charts\"\n",
              "            style=\"display:none;\">\n",
              "\n",
              "<svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\"viewBox=\"0 0 24 24\"\n",
              "     width=\"24px\">\n",
              "    <g>\n",
              "        <path d=\"M19 3H5c-1.1 0-2 .9-2 2v14c0 1.1.9 2 2 2h14c1.1 0 2-.9 2-2V5c0-1.1-.9-2-2-2zM9 17H7v-7h2v7zm4 0h-2V7h2v10zm4 0h-2v-4h2v4z\"/>\n",
              "    </g>\n",
              "</svg>\n",
              "  </button>\n",
              "\n",
              "<style>\n",
              "  .colab-df-quickchart {\n",
              "      --bg-color: #E8F0FE;\n",
              "      --fill-color: #1967D2;\n",
              "      --hover-bg-color: #E2EBFA;\n",
              "      --hover-fill-color: #174EA6;\n",
              "      --disabled-fill-color: #AAA;\n",
              "      --disabled-bg-color: #DDD;\n",
              "  }\n",
              "\n",
              "  [theme=dark] .colab-df-quickchart {\n",
              "      --bg-color: #3B4455;\n",
              "      --fill-color: #D2E3FC;\n",
              "      --hover-bg-color: #434B5C;\n",
              "      --hover-fill-color: #FFFFFF;\n",
              "      --disabled-bg-color: #3B4455;\n",
              "      --disabled-fill-color: #666;\n",
              "  }\n",
              "\n",
              "  .colab-df-quickchart {\n",
              "    background-color: var(--bg-color);\n",
              "    border: none;\n",
              "    border-radius: 50%;\n",
              "    cursor: pointer;\n",
              "    display: none;\n",
              "    fill: var(--fill-color);\n",
              "    height: 32px;\n",
              "    padding: 0;\n",
              "    width: 32px;\n",
              "  }\n",
              "\n",
              "  .colab-df-quickchart:hover {\n",
              "    background-color: var(--hover-bg-color);\n",
              "    box-shadow: 0 1px 2px rgba(60, 64, 67, 0.3), 0 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
              "    fill: var(--button-hover-fill-color);\n",
              "  }\n",
              "\n",
              "  .colab-df-quickchart-complete:disabled,\n",
              "  .colab-df-quickchart-complete:disabled:hover {\n",
              "    background-color: var(--disabled-bg-color);\n",
              "    fill: var(--disabled-fill-color);\n",
              "    box-shadow: none;\n",
              "  }\n",
              "\n",
              "  .colab-df-spinner {\n",
              "    border: 2px solid var(--fill-color);\n",
              "    border-color: transparent;\n",
              "    border-bottom-color: var(--fill-color);\n",
              "    animation:\n",
              "      spin 1s steps(1) infinite;\n",
              "  }\n",
              "\n",
              "  @keyframes spin {\n",
              "    0% {\n",
              "      border-color: transparent;\n",
              "      border-bottom-color: var(--fill-color);\n",
              "      border-left-color: var(--fill-color);\n",
              "    }\n",
              "    20% {\n",
              "      border-color: transparent;\n",
              "      border-left-color: var(--fill-color);\n",
              "      border-top-color: var(--fill-color);\n",
              "    }\n",
              "    30% {\n",
              "      border-color: transparent;\n",
              "      border-left-color: var(--fill-color);\n",
              "      border-top-color: var(--fill-color);\n",
              "      border-right-color: var(--fill-color);\n",
              "    }\n",
              "    40% {\n",
              "      border-color: transparent;\n",
              "      border-right-color: var(--fill-color);\n",
              "      border-top-color: var(--fill-color);\n",
              "    }\n",
              "    60% {\n",
              "      border-color: transparent;\n",
              "      border-right-color: var(--fill-color);\n",
              "    }\n",
              "    80% {\n",
              "      border-color: transparent;\n",
              "      border-right-color: var(--fill-color);\n",
              "      border-bottom-color: var(--fill-color);\n",
              "    }\n",
              "    90% {\n",
              "      border-color: transparent;\n",
              "      border-bottom-color: var(--fill-color);\n",
              "    }\n",
              "  }\n",
              "</style>\n",
              "\n",
              "  <script>\n",
              "    async function quickchart(key) {\n",
              "      const quickchartButtonEl =\n",
              "        document.querySelector('#' + key + ' button');\n",
              "      quickchartButtonEl.disabled = true;  // To prevent multiple clicks.\n",
              "      quickchartButtonEl.classList.add('colab-df-spinner');\n",
              "      try {\n",
              "        const charts = await google.colab.kernel.invokeFunction(\n",
              "            'suggestCharts', [key], {});\n",
              "      } catch (error) {\n",
              "        console.error('Error during call to suggestCharts:', error);\n",
              "      }\n",
              "      quickchartButtonEl.classList.remove('colab-df-spinner');\n",
              "      quickchartButtonEl.classList.add('colab-df-quickchart-complete');\n",
              "    }\n",
              "    (() => {\n",
              "      let quickchartButtonEl =\n",
              "        document.querySelector('#df-46a6bdac-490e-4fb6-99be-12127504cd70 button');\n",
              "      quickchartButtonEl.style.display =\n",
              "        google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
              "    })();\n",
              "  </script>\n",
              "</div>\n",
              "    </div>\n",
              "  </div>\n"
            ],
            "application/vnd.google.colaboratory.intrinsic+json": {
              "type": "dataframe",
              "variable_name": "labels_df",
              "summary": "{\n  \"name\": \"labels_df\",\n  \"rows\": 924,\n  \"fields\": [\n    {\n      \"column\": \"filename\",\n      \"properties\": {\n        \"dtype\": \"category\",\n        \"num_unique_values\": 210,\n        \"samples\": [\n          \"3 (121).jpg\",\n          \"3 (151).jpg\",\n          \"3 (111).jpg\"\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"width\",\n      \"properties\": {\n        \"dtype\": \"number\",\n        \"std\": 0,\n        \"min\": 854,\n        \"max\": 854,\n        \"num_unique_values\": 1,\n        \"samples\": [\n          854\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"height\",\n      \"properties\": {\n        \"dtype\": \"number\",\n        \"std\": 0,\n        \"min\": 480,\n        \"max\": 480,\n        \"num_unique_values\": 1,\n        \"samples\": [\n          480\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"class\",\n      \"properties\": {\n        \"dtype\": \"category\",\n        \"num_unique_values\": 3,\n        \"samples\": [\n          \"break\"\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"xmin\",\n      \"properties\": {\n        \"dtype\": \"number\",\n        \"std\": 161.20046445122384,\n        \"min\": 124.4,\n        \"max\": 695.3,\n        \"num_unique_values\": 267,\n        \"samples\": [\n          344.5\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"ymin\",\n      \"properties\": {\n        \"dtype\": \"number\",\n        \"std\": 138.3967558644383,\n        \"min\": 0.0,\n        \"max\": 408.27,\n        \"num_unique_values\": 229,\n        \"samples\": [\n          82.5\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"xmax\",\n      \"properties\": {\n        \"dtype\": \"number\",\n        \"std\": 157.95659769506008,\n        \"min\": 142.09,\n        \"max\": 729.2,\n        \"num_unique_values\": 274,\n        \"samples\": [\n          277.33\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"ymax\",\n      \"properties\": {\n        \"dtype\": \"number\",\n        \"std\": 88.52809947274022,\n        \"min\": 134.2,\n        \"max\": 480.0,\n        \"num_unique_values\": 256,\n        \"samples\": [\n          289.51\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    }\n  ]\n}"
            }
          },
          "metadata": {},
          "execution_count": 52
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "# Укажите исходные и новые размеры изображений\n",
        "original_width, original_height = 854, 480\n",
        "new_width, new_height = 600, 337"
      ],
      "metadata": {
        "id": "Ak7pFA1bliZC"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def resize_image_and_update_coords(image_path, boxes):\n",
        "    with Image.open(image_path) as img:\n",
        "        img_resized = img.resize((new_width, new_height), Image.ANTIALIAS)\n",
        "\n",
        "        # Сохраните новое изображение в нужную директорию (например, \"images_resized\")\n",
        "        new_image_path = os.path.join(\"images_resized\", os.path.basename(image_path))\n",
        "        os.makedirs(\"images_resized\", exist_ok=True)\n",
        "        img_resized.save(new_image_path)\n",
        "\n",
        "        # Обновите координаты bounding boxes\n",
        "        scale_x = new_width / original_width\n",
        "        scale_y = new_height / original_height\n",
        "        boxes = boxes.copy()\n",
        "        boxes['xmin'] = boxes['xmin'] * scale_x\n",
        "        boxes['ymin'] = boxes['ymin'] * scale_y\n",
        "        boxes['xmax'] = boxes['xmax'] * scale_x\n",
        "        boxes['ymax'] = boxes['ymax'] * scale_y\n",
        "\n",
        "        # Обновите размеры изображения в DataFrame\n",
        "        boxes['width'] = new_width\n",
        "        boxes['height'] = new_height\n",
        "        boxes['filename'] = os.path.basename(new_image_path)\n",
        "\n",
        "        return boxes"
      ],
      "metadata": {
        "id": "MnuUR9LhBbWN"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "# Обрабатываем каждое изображение и обновляем координаты\n",
        "updated_frames = []\n",
        "for image_file in tqdm(labels_df['filename'].unique()):\n",
        "    image_path = os.path.join(\"images\", image_file)  # Укажите путь к директории с изображениями\n",
        "\n",
        "    # Получаем все записи для данного изображения\n",
        "    image_boxes = labels_df[labels_df['filename'] == image_file]\n",
        "\n",
        "    # Обновляем координаты bounding boxes\n",
        "    updated_boxes = resize_image_and_update_coords(image_path, image_boxes)\n",
        "\n",
        "    updated_frames.append(updated_boxes)\n",
        "\n",
        "# Объединяем все обновленные DataFrame\n",
        "updated_labels_df = pd.concat(updated_frames)"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "asiMyeKoBbTk",
        "outputId": "b639678b-69d7-43db-b086-becc7a5b58ae"
      },
      "execution_count": null,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "  0%|          | 0/210 [00:00<?, ?it/s]<ipython-input-54-4ea363e3c06b>:3: DeprecationWarning: ANTIALIAS is deprecated and will be removed in Pillow 10 (2023-07-01). Use LANCZOS or Resampling.LANCZOS instead.\n",
            "  img_resized = img.resize((new_width, new_height), Image.ANTIALIAS)\n",
            "100%|██████████| 210/210 [00:03<00:00, 55.38it/s]\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "updated_labels_df"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 424
        },
        "id": "VPs0f1u1BbQm",
        "outputId": "ce790b23-3a12-4db1-df6a-826dd319b817"
      },
      "execution_count": null,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "        filename  width  height  class        xmin        ymin        xmax  \\\n",
              "0    2 (112).jpg    600     337  break  355.166276  124.353000  377.339578   \n",
              "1    2 (112).jpg    600     337  crack   89.030445   27.416354  385.538642   \n",
              "2    2 (112).jpg    600     337  joint  356.522248    0.842500  507.786885   \n",
              "3    2 (112).jpg    600     337  crack  444.449649    1.965833  509.859485   \n",
              "462  2 (112).jpg    600     337  break  355.166276  124.353000  377.339578   \n",
              "..           ...    ...     ...    ...         ...         ...         ...   \n",
              "461  2 (114).jpg    600     337  crack  465.021077    1.874562  510.281030   \n",
              "920  2 (114).jpg    600     337  break  356.444965  122.843521  378.604215   \n",
              "921  2 (114).jpg    600     337  crack   89.030445   26.117500  390.351288   \n",
              "922  2 (114).jpg    600     337  joint  360.562061    1.333958  511.545667   \n",
              "923  2 (114).jpg    600     337  crack  465.021077    1.874562  510.281030   \n",
              "\n",
              "           ymax  \n",
              "0    163.599458  \n",
              "1    320.571250  \n",
              "2    336.859583  \n",
              "3    247.695000  \n",
              "462  163.599458  \n",
              "..          ...  \n",
              "461  208.308125  \n",
              "920  157.561542  \n",
              "921  320.571250  \n",
              "922  337.000000  \n",
              "923  208.308125  \n",
              "\n",
              "[924 rows x 8 columns]"
            ],
            "text/html": [
              "\n",
              "  <div id=\"df-2f5629ad-eb8d-4c4b-a51c-ad8e048f54a9\" class=\"colab-df-container\">\n",
              "    <div>\n",
              "<style scoped>\n",
              "    .dataframe tbody tr th:only-of-type {\n",
              "        vertical-align: middle;\n",
              "    }\n",
              "\n",
              "    .dataframe tbody tr th {\n",
              "        vertical-align: top;\n",
              "    }\n",
              "\n",
              "    .dataframe thead th {\n",
              "        text-align: right;\n",
              "    }\n",
              "</style>\n",
              "<table border=\"1\" class=\"dataframe\">\n",
              "  <thead>\n",
              "    <tr style=\"text-align: right;\">\n",
              "      <th></th>\n",
              "      <th>filename</th>\n",
              "      <th>width</th>\n",
              "      <th>height</th>\n",
              "      <th>class</th>\n",
              "      <th>xmin</th>\n",
              "      <th>ymin</th>\n",
              "      <th>xmax</th>\n",
              "      <th>ymax</th>\n",
              "    </tr>\n",
              "  </thead>\n",
              "  <tbody>\n",
              "    <tr>\n",
              "      <th>0</th>\n",
              "      <td>2 (112).jpg</td>\n",
              "      <td>600</td>\n",
              "      <td>337</td>\n",
              "      <td>break</td>\n",
              "      <td>355.166276</td>\n",
              "      <td>124.353000</td>\n",
              "      <td>377.339578</td>\n",
              "      <td>163.599458</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>1</th>\n",
              "      <td>2 (112).jpg</td>\n",
              "      <td>600</td>\n",
              "      <td>337</td>\n",
              "      <td>crack</td>\n",
              "      <td>89.030445</td>\n",
              "      <td>27.416354</td>\n",
              "      <td>385.538642</td>\n",
              "      <td>320.571250</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>2</th>\n",
              "      <td>2 (112).jpg</td>\n",
              "      <td>600</td>\n",
              "      <td>337</td>\n",
              "      <td>joint</td>\n",
              "      <td>356.522248</td>\n",
              "      <td>0.842500</td>\n",
              "      <td>507.786885</td>\n",
              "      <td>336.859583</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>3</th>\n",
              "      <td>2 (112).jpg</td>\n",
              "      <td>600</td>\n",
              "      <td>337</td>\n",
              "      <td>crack</td>\n",
              "      <td>444.449649</td>\n",
              "      <td>1.965833</td>\n",
              "      <td>509.859485</td>\n",
              "      <td>247.695000</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>462</th>\n",
              "      <td>2 (112).jpg</td>\n",
              "      <td>600</td>\n",
              "      <td>337</td>\n",
              "      <td>break</td>\n",
              "      <td>355.166276</td>\n",
              "      <td>124.353000</td>\n",
              "      <td>377.339578</td>\n",
              "      <td>163.599458</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>...</th>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>461</th>\n",
              "      <td>2 (114).jpg</td>\n",
              "      <td>600</td>\n",
              "      <td>337</td>\n",
              "      <td>crack</td>\n",
              "      <td>465.021077</td>\n",
              "      <td>1.874562</td>\n",
              "      <td>510.281030</td>\n",
              "      <td>208.308125</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>920</th>\n",
              "      <td>2 (114).jpg</td>\n",
              "      <td>600</td>\n",
              "      <td>337</td>\n",
              "      <td>break</td>\n",
              "      <td>356.444965</td>\n",
              "      <td>122.843521</td>\n",
              "      <td>378.604215</td>\n",
              "      <td>157.561542</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>921</th>\n",
              "      <td>2 (114).jpg</td>\n",
              "      <td>600</td>\n",
              "      <td>337</td>\n",
              "      <td>crack</td>\n",
              "      <td>89.030445</td>\n",
              "      <td>26.117500</td>\n",
              "      <td>390.351288</td>\n",
              "      <td>320.571250</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>922</th>\n",
              "      <td>2 (114).jpg</td>\n",
              "      <td>600</td>\n",
              "      <td>337</td>\n",
              "      <td>joint</td>\n",
              "      <td>360.562061</td>\n",
              "      <td>1.333958</td>\n",
              "      <td>511.545667</td>\n",
              "      <td>337.000000</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>923</th>\n",
              "      <td>2 (114).jpg</td>\n",
              "      <td>600</td>\n",
              "      <td>337</td>\n",
              "      <td>crack</td>\n",
              "      <td>465.021077</td>\n",
              "      <td>1.874562</td>\n",
              "      <td>510.281030</td>\n",
              "      <td>208.308125</td>\n",
              "    </tr>\n",
              "  </tbody>\n",
              "</table>\n",
              "<p>924 rows × 8 columns</p>\n",
              "</div>\n",
              "    <div class=\"colab-df-buttons\">\n",
              "\n",
              "  <div class=\"colab-df-container\">\n",
              "    <button class=\"colab-df-convert\" onclick=\"convertToInteractive('df-2f5629ad-eb8d-4c4b-a51c-ad8e048f54a9')\"\n",
              "            title=\"Convert this dataframe to an interactive table.\"\n",
              "            style=\"display:none;\">\n",
              "\n",
              "  <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\" viewBox=\"0 -960 960 960\">\n",
              "    <path d=\"M120-120v-720h720v720H120Zm60-500h600v-160H180v160Zm220 220h160v-160H400v160Zm0 220h160v-160H400v160ZM180-400h160v-160H180v160Zm440 0h160v-160H620v160ZM180-180h160v-160H180v160Zm440 0h160v-160H620v160Z\"/>\n",
              "  </svg>\n",
              "    </button>\n",
              "\n",
              "  <style>\n",
              "    .colab-df-container {\n",
              "      display:flex;\n",
              "      gap: 12px;\n",
              "    }\n",
              "\n",
              "    .colab-df-convert {\n",
              "      background-color: #E8F0FE;\n",
              "      border: none;\n",
              "      border-radius: 50%;\n",
              "      cursor: pointer;\n",
              "      display: none;\n",
              "      fill: #1967D2;\n",
              "      height: 32px;\n",
              "      padding: 0 0 0 0;\n",
              "      width: 32px;\n",
              "    }\n",
              "\n",
              "    .colab-df-convert:hover {\n",
              "      background-color: #E2EBFA;\n",
              "      box-shadow: 0px 1px 2px rgba(60, 64, 67, 0.3), 0px 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
              "      fill: #174EA6;\n",
              "    }\n",
              "\n",
              "    .colab-df-buttons div {\n",
              "      margin-bottom: 4px;\n",
              "    }\n",
              "\n",
              "    [theme=dark] .colab-df-convert {\n",
              "      background-color: #3B4455;\n",
              "      fill: #D2E3FC;\n",
              "    }\n",
              "\n",
              "    [theme=dark] .colab-df-convert:hover {\n",
              "      background-color: #434B5C;\n",
              "      box-shadow: 0px 1px 3px 1px rgba(0, 0, 0, 0.15);\n",
              "      filter: drop-shadow(0px 1px 2px rgba(0, 0, 0, 0.3));\n",
              "      fill: #FFFFFF;\n",
              "    }\n",
              "  </style>\n",
              "\n",
              "    <script>\n",
              "      const buttonEl =\n",
              "        document.querySelector('#df-2f5629ad-eb8d-4c4b-a51c-ad8e048f54a9 button.colab-df-convert');\n",
              "      buttonEl.style.display =\n",
              "        google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
              "\n",
              "      async function convertToInteractive(key) {\n",
              "        const element = document.querySelector('#df-2f5629ad-eb8d-4c4b-a51c-ad8e048f54a9');\n",
              "        const dataTable =\n",
              "          await google.colab.kernel.invokeFunction('convertToInteractive',\n",
              "                                                    [key], {});\n",
              "        if (!dataTable) return;\n",
              "\n",
              "        const docLinkHtml = 'Like what you see? Visit the ' +\n",
              "          '<a target=\"_blank\" href=https://colab.research.google.com/notebooks/data_table.ipynb>data table notebook</a>'\n",
              "          + ' to learn more about interactive tables.';\n",
              "        element.innerHTML = '';\n",
              "        dataTable['output_type'] = 'display_data';\n",
              "        await google.colab.output.renderOutput(dataTable, element);\n",
              "        const docLink = document.createElement('div');\n",
              "        docLink.innerHTML = docLinkHtml;\n",
              "        element.appendChild(docLink);\n",
              "      }\n",
              "    </script>\n",
              "  </div>\n",
              "\n",
              "\n",
              "<div id=\"df-07388e50-7b80-4ae7-bf6a-15d824abae80\">\n",
              "  <button class=\"colab-df-quickchart\" onclick=\"quickchart('df-07388e50-7b80-4ae7-bf6a-15d824abae80')\"\n",
              "            title=\"Suggest charts\"\n",
              "            style=\"display:none;\">\n",
              "\n",
              "<svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\"viewBox=\"0 0 24 24\"\n",
              "     width=\"24px\">\n",
              "    <g>\n",
              "        <path d=\"M19 3H5c-1.1 0-2 .9-2 2v14c0 1.1.9 2 2 2h14c1.1 0 2-.9 2-2V5c0-1.1-.9-2-2-2zM9 17H7v-7h2v7zm4 0h-2V7h2v10zm4 0h-2v-4h2v4z\"/>\n",
              "    </g>\n",
              "</svg>\n",
              "  </button>\n",
              "\n",
              "<style>\n",
              "  .colab-df-quickchart {\n",
              "      --bg-color: #E8F0FE;\n",
              "      --fill-color: #1967D2;\n",
              "      --hover-bg-color: #E2EBFA;\n",
              "      --hover-fill-color: #174EA6;\n",
              "      --disabled-fill-color: #AAA;\n",
              "      --disabled-bg-color: #DDD;\n",
              "  }\n",
              "\n",
              "  [theme=dark] .colab-df-quickchart {\n",
              "      --bg-color: #3B4455;\n",
              "      --fill-color: #D2E3FC;\n",
              "      --hover-bg-color: #434B5C;\n",
              "      --hover-fill-color: #FFFFFF;\n",
              "      --disabled-bg-color: #3B4455;\n",
              "      --disabled-fill-color: #666;\n",
              "  }\n",
              "\n",
              "  .colab-df-quickchart {\n",
              "    background-color: var(--bg-color);\n",
              "    border: none;\n",
              "    border-radius: 50%;\n",
              "    cursor: pointer;\n",
              "    display: none;\n",
              "    fill: var(--fill-color);\n",
              "    height: 32px;\n",
              "    padding: 0;\n",
              "    width: 32px;\n",
              "  }\n",
              "\n",
              "  .colab-df-quickchart:hover {\n",
              "    background-color: var(--hover-bg-color);\n",
              "    box-shadow: 0 1px 2px rgba(60, 64, 67, 0.3), 0 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
              "    fill: var(--button-hover-fill-color);\n",
              "  }\n",
              "\n",
              "  .colab-df-quickchart-complete:disabled,\n",
              "  .colab-df-quickchart-complete:disabled:hover {\n",
              "    background-color: var(--disabled-bg-color);\n",
              "    fill: var(--disabled-fill-color);\n",
              "    box-shadow: none;\n",
              "  }\n",
              "\n",
              "  .colab-df-spinner {\n",
              "    border: 2px solid var(--fill-color);\n",
              "    border-color: transparent;\n",
              "    border-bottom-color: var(--fill-color);\n",
              "    animation:\n",
              "      spin 1s steps(1) infinite;\n",
              "  }\n",
              "\n",
              "  @keyframes spin {\n",
              "    0% {\n",
              "      border-color: transparent;\n",
              "      border-bottom-color: var(--fill-color);\n",
              "      border-left-color: var(--fill-color);\n",
              "    }\n",
              "    20% {\n",
              "      border-color: transparent;\n",
              "      border-left-color: var(--fill-color);\n",
              "      border-top-color: var(--fill-color);\n",
              "    }\n",
              "    30% {\n",
              "      border-color: transparent;\n",
              "      border-left-color: var(--fill-color);\n",
              "      border-top-color: var(--fill-color);\n",
              "      border-right-color: var(--fill-color);\n",
              "    }\n",
              "    40% {\n",
              "      border-color: transparent;\n",
              "      border-right-color: var(--fill-color);\n",
              "      border-top-color: var(--fill-color);\n",
              "    }\n",
              "    60% {\n",
              "      border-color: transparent;\n",
              "      border-right-color: var(--fill-color);\n",
              "    }\n",
              "    80% {\n",
              "      border-color: transparent;\n",
              "      border-right-color: var(--fill-color);\n",
              "      border-bottom-color: var(--fill-color);\n",
              "    }\n",
              "    90% {\n",
              "      border-color: transparent;\n",
              "      border-bottom-color: var(--fill-color);\n",
              "    }\n",
              "  }\n",
              "</style>\n",
              "\n",
              "  <script>\n",
              "    async function quickchart(key) {\n",
              "      const quickchartButtonEl =\n",
              "        document.querySelector('#' + key + ' button');\n",
              "      quickchartButtonEl.disabled = true;  // To prevent multiple clicks.\n",
              "      quickchartButtonEl.classList.add('colab-df-spinner');\n",
              "      try {\n",
              "        const charts = await google.colab.kernel.invokeFunction(\n",
              "            'suggestCharts', [key], {});\n",
              "      } catch (error) {\n",
              "        console.error('Error during call to suggestCharts:', error);\n",
              "      }\n",
              "      quickchartButtonEl.classList.remove('colab-df-spinner');\n",
              "      quickchartButtonEl.classList.add('colab-df-quickchart-complete');\n",
              "    }\n",
              "    (() => {\n",
              "      let quickchartButtonEl =\n",
              "        document.querySelector('#df-07388e50-7b80-4ae7-bf6a-15d824abae80 button');\n",
              "      quickchartButtonEl.style.display =\n",
              "        google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
              "    })();\n",
              "  </script>\n",
              "</div>\n",
              "    </div>\n",
              "  </div>\n"
            ],
            "application/vnd.google.colaboratory.intrinsic+json": {
              "type": "dataframe",
              "variable_name": "updated_labels_df",
              "summary": "{\n  \"name\": \"updated_labels_df\",\n  \"rows\": 924,\n  \"fields\": [\n    {\n      \"column\": \"filename\",\n      \"properties\": {\n        \"dtype\": \"category\",\n        \"num_unique_values\": 210,\n        \"samples\": [\n          \"3 (121).jpg\",\n          \"3 (151).jpg\",\n          \"3 (111).jpg\"\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"width\",\n      \"properties\": {\n        \"dtype\": \"number\",\n        \"std\": 0,\n        \"min\": 600,\n        \"max\": 600,\n        \"num_unique_values\": 1,\n        \"samples\": [\n          600\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"height\",\n      \"properties\": {\n        \"dtype\": \"number\",\n        \"std\": 0,\n        \"min\": 337,\n        \"max\": 337,\n        \"num_unique_values\": 1,\n        \"samples\": [\n          337\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"class\",\n      \"properties\": {\n        \"dtype\": \"category\",\n        \"num_unique_values\": 3,\n        \"samples\": [\n          \"break\"\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"xmin\",\n      \"properties\": {\n        \"dtype\": \"number\",\n        \"std\": 113.25559563317836,\n        \"min\": 87.40046838407494,\n        \"max\": 488.5011709601873,\n        \"num_unique_values\": 267,\n        \"samples\": [\n          242.03747072599532\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"ymin\",\n      \"properties\": {\n        \"dtype\": \"number\",\n        \"std\": 97.1660556798244,\n        \"min\": 0.0,\n        \"max\": 286.63956249999995,\n        \"num_unique_values\": 229,\n        \"samples\": [\n          57.92187499999999\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"xmax\",\n      \"properties\": {\n        \"dtype\": \"number\",\n        \"std\": 110.97653233844972,\n        \"min\": 99.82903981264637,\n        \"max\": 512.3185011709602,\n        \"num_unique_values\": 274,\n        \"samples\": [\n          194.8454332552693\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"ymax\",\n      \"properties\": {\n        \"dtype\": \"number\",\n        \"std\": 62.15410317148637,\n        \"min\": 94.21958333333332,\n        \"max\": 337.0,\n        \"num_unique_values\": 256,\n        \"samples\": [\n          203.26014583333333\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    }\n  ]\n}"
            }
          },
          "metadata": {},
          "execution_count": 57
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "from sklearn.preprocessing import LabelEncoder\n",
        "\n",
        "labelencoder = LabelEncoder()\n",
        "updated_labels_df['class'] = labelencoder.fit_transform(updated_labels_df['class'])\n"
      ],
      "metadata": {
        "id": "-m0TVvYnz-Jl"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "updated_labels_df"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 424
        },
        "id": "TVD_SDJ60SUV",
        "outputId": "2d26053d-c493-49c8-e5a1-cb99de681ec4"
      },
      "execution_count": null,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "        filename  width  height  class        xmin        ymin        xmax  \\\n",
              "0    2 (112).jpg    600     337      0  355.166276  124.353000  377.339578   \n",
              "1    2 (112).jpg    600     337      1   89.030445   27.416354  385.538642   \n",
              "2    2 (112).jpg    600     337      2  356.522248    0.842500  507.786885   \n",
              "3    2 (112).jpg    600     337      1  444.449649    1.965833  509.859485   \n",
              "462  2 (112).jpg    600     337      0  355.166276  124.353000  377.339578   \n",
              "..           ...    ...     ...    ...         ...         ...         ...   \n",
              "461  2 (114).jpg    600     337      1  465.021077    1.874562  510.281030   \n",
              "920  2 (114).jpg    600     337      0  356.444965  122.843521  378.604215   \n",
              "921  2 (114).jpg    600     337      1   89.030445   26.117500  390.351288   \n",
              "922  2 (114).jpg    600     337      2  360.562061    1.333958  511.545667   \n",
              "923  2 (114).jpg    600     337      1  465.021077    1.874562  510.281030   \n",
              "\n",
              "           ymax  \n",
              "0    163.599458  \n",
              "1    320.571250  \n",
              "2    336.859583  \n",
              "3    247.695000  \n",
              "462  163.599458  \n",
              "..          ...  \n",
              "461  208.308125  \n",
              "920  157.561542  \n",
              "921  320.571250  \n",
              "922  337.000000  \n",
              "923  208.308125  \n",
              "\n",
              "[924 rows x 8 columns]"
            ],
            "text/html": [
              "\n",
              "  <div id=\"df-5d6f6ab7-5462-4ad9-a536-4d2690be3b94\" class=\"colab-df-container\">\n",
              "    <div>\n",
              "<style scoped>\n",
              "    .dataframe tbody tr th:only-of-type {\n",
              "        vertical-align: middle;\n",
              "    }\n",
              "\n",
              "    .dataframe tbody tr th {\n",
              "        vertical-align: top;\n",
              "    }\n",
              "\n",
              "    .dataframe thead th {\n",
              "        text-align: right;\n",
              "    }\n",
              "</style>\n",
              "<table border=\"1\" class=\"dataframe\">\n",
              "  <thead>\n",
              "    <tr style=\"text-align: right;\">\n",
              "      <th></th>\n",
              "      <th>filename</th>\n",
              "      <th>width</th>\n",
              "      <th>height</th>\n",
              "      <th>class</th>\n",
              "      <th>xmin</th>\n",
              "      <th>ymin</th>\n",
              "      <th>xmax</th>\n",
              "      <th>ymax</th>\n",
              "    </tr>\n",
              "  </thead>\n",
              "  <tbody>\n",
              "    <tr>\n",
              "      <th>0</th>\n",
              "      <td>2 (112).jpg</td>\n",
              "      <td>600</td>\n",
              "      <td>337</td>\n",
              "      <td>0</td>\n",
              "      <td>355.166276</td>\n",
              "      <td>124.353000</td>\n",
              "      <td>377.339578</td>\n",
              "      <td>163.599458</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>1</th>\n",
              "      <td>2 (112).jpg</td>\n",
              "      <td>600</td>\n",
              "      <td>337</td>\n",
              "      <td>1</td>\n",
              "      <td>89.030445</td>\n",
              "      <td>27.416354</td>\n",
              "      <td>385.538642</td>\n",
              "      <td>320.571250</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>2</th>\n",
              "      <td>2 (112).jpg</td>\n",
              "      <td>600</td>\n",
              "      <td>337</td>\n",
              "      <td>2</td>\n",
              "      <td>356.522248</td>\n",
              "      <td>0.842500</td>\n",
              "      <td>507.786885</td>\n",
              "      <td>336.859583</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>3</th>\n",
              "      <td>2 (112).jpg</td>\n",
              "      <td>600</td>\n",
              "      <td>337</td>\n",
              "      <td>1</td>\n",
              "      <td>444.449649</td>\n",
              "      <td>1.965833</td>\n",
              "      <td>509.859485</td>\n",
              "      <td>247.695000</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>462</th>\n",
              "      <td>2 (112).jpg</td>\n",
              "      <td>600</td>\n",
              "      <td>337</td>\n",
              "      <td>0</td>\n",
              "      <td>355.166276</td>\n",
              "      <td>124.353000</td>\n",
              "      <td>377.339578</td>\n",
              "      <td>163.599458</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>...</th>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>461</th>\n",
              "      <td>2 (114).jpg</td>\n",
              "      <td>600</td>\n",
              "      <td>337</td>\n",
              "      <td>1</td>\n",
              "      <td>465.021077</td>\n",
              "      <td>1.874562</td>\n",
              "      <td>510.281030</td>\n",
              "      <td>208.308125</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>920</th>\n",
              "      <td>2 (114).jpg</td>\n",
              "      <td>600</td>\n",
              "      <td>337</td>\n",
              "      <td>0</td>\n",
              "      <td>356.444965</td>\n",
              "      <td>122.843521</td>\n",
              "      <td>378.604215</td>\n",
              "      <td>157.561542</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>921</th>\n",
              "      <td>2 (114).jpg</td>\n",
              "      <td>600</td>\n",
              "      <td>337</td>\n",
              "      <td>1</td>\n",
              "      <td>89.030445</td>\n",
              "      <td>26.117500</td>\n",
              "      <td>390.351288</td>\n",
              "      <td>320.571250</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>922</th>\n",
              "      <td>2 (114).jpg</td>\n",
              "      <td>600</td>\n",
              "      <td>337</td>\n",
              "      <td>2</td>\n",
              "      <td>360.562061</td>\n",
              "      <td>1.333958</td>\n",
              "      <td>511.545667</td>\n",
              "      <td>337.000000</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>923</th>\n",
              "      <td>2 (114).jpg</td>\n",
              "      <td>600</td>\n",
              "      <td>337</td>\n",
              "      <td>1</td>\n",
              "      <td>465.021077</td>\n",
              "      <td>1.874562</td>\n",
              "      <td>510.281030</td>\n",
              "      <td>208.308125</td>\n",
              "    </tr>\n",
              "  </tbody>\n",
              "</table>\n",
              "<p>924 rows × 8 columns</p>\n",
              "</div>\n",
              "    <div class=\"colab-df-buttons\">\n",
              "\n",
              "  <div class=\"colab-df-container\">\n",
              "    <button class=\"colab-df-convert\" onclick=\"convertToInteractive('df-5d6f6ab7-5462-4ad9-a536-4d2690be3b94')\"\n",
              "            title=\"Convert this dataframe to an interactive table.\"\n",
              "            style=\"display:none;\">\n",
              "\n",
              "  <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\" viewBox=\"0 -960 960 960\">\n",
              "    <path d=\"M120-120v-720h720v720H120Zm60-500h600v-160H180v160Zm220 220h160v-160H400v160Zm0 220h160v-160H400v160ZM180-400h160v-160H180v160Zm440 0h160v-160H620v160ZM180-180h160v-160H180v160Zm440 0h160v-160H620v160Z\"/>\n",
              "  </svg>\n",
              "    </button>\n",
              "\n",
              "  <style>\n",
              "    .colab-df-container {\n",
              "      display:flex;\n",
              "      gap: 12px;\n",
              "    }\n",
              "\n",
              "    .colab-df-convert {\n",
              "      background-color: #E8F0FE;\n",
              "      border: none;\n",
              "      border-radius: 50%;\n",
              "      cursor: pointer;\n",
              "      display: none;\n",
              "      fill: #1967D2;\n",
              "      height: 32px;\n",
              "      padding: 0 0 0 0;\n",
              "      width: 32px;\n",
              "    }\n",
              "\n",
              "    .colab-df-convert:hover {\n",
              "      background-color: #E2EBFA;\n",
              "      box-shadow: 0px 1px 2px rgba(60, 64, 67, 0.3), 0px 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
              "      fill: #174EA6;\n",
              "    }\n",
              "\n",
              "    .colab-df-buttons div {\n",
              "      margin-bottom: 4px;\n",
              "    }\n",
              "\n",
              "    [theme=dark] .colab-df-convert {\n",
              "      background-color: #3B4455;\n",
              "      fill: #D2E3FC;\n",
              "    }\n",
              "\n",
              "    [theme=dark] .colab-df-convert:hover {\n",
              "      background-color: #434B5C;\n",
              "      box-shadow: 0px 1px 3px 1px rgba(0, 0, 0, 0.15);\n",
              "      filter: drop-shadow(0px 1px 2px rgba(0, 0, 0, 0.3));\n",
              "      fill: #FFFFFF;\n",
              "    }\n",
              "  </style>\n",
              "\n",
              "    <script>\n",
              "      const buttonEl =\n",
              "        document.querySelector('#df-5d6f6ab7-5462-4ad9-a536-4d2690be3b94 button.colab-df-convert');\n",
              "      buttonEl.style.display =\n",
              "        google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
              "\n",
              "      async function convertToInteractive(key) {\n",
              "        const element = document.querySelector('#df-5d6f6ab7-5462-4ad9-a536-4d2690be3b94');\n",
              "        const dataTable =\n",
              "          await google.colab.kernel.invokeFunction('convertToInteractive',\n",
              "                                                    [key], {});\n",
              "        if (!dataTable) return;\n",
              "\n",
              "        const docLinkHtml = 'Like what you see? Visit the ' +\n",
              "          '<a target=\"_blank\" href=https://colab.research.google.com/notebooks/data_table.ipynb>data table notebook</a>'\n",
              "          + ' to learn more about interactive tables.';\n",
              "        element.innerHTML = '';\n",
              "        dataTable['output_type'] = 'display_data';\n",
              "        await google.colab.output.renderOutput(dataTable, element);\n",
              "        const docLink = document.createElement('div');\n",
              "        docLink.innerHTML = docLinkHtml;\n",
              "        element.appendChild(docLink);\n",
              "      }\n",
              "    </script>\n",
              "  </div>\n",
              "\n",
              "\n",
              "<div id=\"df-19376ce3-9257-4eaf-955f-e0df3d165400\">\n",
              "  <button class=\"colab-df-quickchart\" onclick=\"quickchart('df-19376ce3-9257-4eaf-955f-e0df3d165400')\"\n",
              "            title=\"Suggest charts\"\n",
              "            style=\"display:none;\">\n",
              "\n",
              "<svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\"viewBox=\"0 0 24 24\"\n",
              "     width=\"24px\">\n",
              "    <g>\n",
              "        <path d=\"M19 3H5c-1.1 0-2 .9-2 2v14c0 1.1.9 2 2 2h14c1.1 0 2-.9 2-2V5c0-1.1-.9-2-2-2zM9 17H7v-7h2v7zm4 0h-2V7h2v10zm4 0h-2v-4h2v4z\"/>\n",
              "    </g>\n",
              "</svg>\n",
              "  </button>\n",
              "\n",
              "<style>\n",
              "  .colab-df-quickchart {\n",
              "      --bg-color: #E8F0FE;\n",
              "      --fill-color: #1967D2;\n",
              "      --hover-bg-color: #E2EBFA;\n",
              "      --hover-fill-color: #174EA6;\n",
              "      --disabled-fill-color: #AAA;\n",
              "      --disabled-bg-color: #DDD;\n",
              "  }\n",
              "\n",
              "  [theme=dark] .colab-df-quickchart {\n",
              "      --bg-color: #3B4455;\n",
              "      --fill-color: #D2E3FC;\n",
              "      --hover-bg-color: #434B5C;\n",
              "      --hover-fill-color: #FFFFFF;\n",
              "      --disabled-bg-color: #3B4455;\n",
              "      --disabled-fill-color: #666;\n",
              "  }\n",
              "\n",
              "  .colab-df-quickchart {\n",
              "    background-color: var(--bg-color);\n",
              "    border: none;\n",
              "    border-radius: 50%;\n",
              "    cursor: pointer;\n",
              "    display: none;\n",
              "    fill: var(--fill-color);\n",
              "    height: 32px;\n",
              "    padding: 0;\n",
              "    width: 32px;\n",
              "  }\n",
              "\n",
              "  .colab-df-quickchart:hover {\n",
              "    background-color: var(--hover-bg-color);\n",
              "    box-shadow: 0 1px 2px rgba(60, 64, 67, 0.3), 0 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
              "    fill: var(--button-hover-fill-color);\n",
              "  }\n",
              "\n",
              "  .colab-df-quickchart-complete:disabled,\n",
              "  .colab-df-quickchart-complete:disabled:hover {\n",
              "    background-color: var(--disabled-bg-color);\n",
              "    fill: var(--disabled-fill-color);\n",
              "    box-shadow: none;\n",
              "  }\n",
              "\n",
              "  .colab-df-spinner {\n",
              "    border: 2px solid var(--fill-color);\n",
              "    border-color: transparent;\n",
              "    border-bottom-color: var(--fill-color);\n",
              "    animation:\n",
              "      spin 1s steps(1) infinite;\n",
              "  }\n",
              "\n",
              "  @keyframes spin {\n",
              "    0% {\n",
              "      border-color: transparent;\n",
              "      border-bottom-color: var(--fill-color);\n",
              "      border-left-color: var(--fill-color);\n",
              "    }\n",
              "    20% {\n",
              "      border-color: transparent;\n",
              "      border-left-color: var(--fill-color);\n",
              "      border-top-color: var(--fill-color);\n",
              "    }\n",
              "    30% {\n",
              "      border-color: transparent;\n",
              "      border-left-color: var(--fill-color);\n",
              "      border-top-color: var(--fill-color);\n",
              "      border-right-color: var(--fill-color);\n",
              "    }\n",
              "    40% {\n",
              "      border-color: transparent;\n",
              "      border-right-color: var(--fill-color);\n",
              "      border-top-color: var(--fill-color);\n",
              "    }\n",
              "    60% {\n",
              "      border-color: transparent;\n",
              "      border-right-color: var(--fill-color);\n",
              "    }\n",
              "    80% {\n",
              "      border-color: transparent;\n",
              "      border-right-color: var(--fill-color);\n",
              "      border-bottom-color: var(--fill-color);\n",
              "    }\n",
              "    90% {\n",
              "      border-color: transparent;\n",
              "      border-bottom-color: var(--fill-color);\n",
              "    }\n",
              "  }\n",
              "</style>\n",
              "\n",
              "  <script>\n",
              "    async function quickchart(key) {\n",
              "      const quickchartButtonEl =\n",
              "        document.querySelector('#' + key + ' button');\n",
              "      quickchartButtonEl.disabled = true;  // To prevent multiple clicks.\n",
              "      quickchartButtonEl.classList.add('colab-df-spinner');\n",
              "      try {\n",
              "        const charts = await google.colab.kernel.invokeFunction(\n",
              "            'suggestCharts', [key], {});\n",
              "      } catch (error) {\n",
              "        console.error('Error during call to suggestCharts:', error);\n",
              "      }\n",
              "      quickchartButtonEl.classList.remove('colab-df-spinner');\n",
              "      quickchartButtonEl.classList.add('colab-df-quickchart-complete');\n",
              "    }\n",
              "    (() => {\n",
              "      let quickchartButtonEl =\n",
              "        document.querySelector('#df-19376ce3-9257-4eaf-955f-e0df3d165400 button');\n",
              "      quickchartButtonEl.style.display =\n",
              "        google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
              "    })();\n",
              "  </script>\n",
              "</div>\n",
              "    </div>\n",
              "  </div>\n"
            ],
            "application/vnd.google.colaboratory.intrinsic+json": {
              "type": "dataframe",
              "variable_name": "updated_labels_df",
              "summary": "{\n  \"name\": \"updated_labels_df\",\n  \"rows\": 924,\n  \"fields\": [\n    {\n      \"column\": \"filename\",\n      \"properties\": {\n        \"dtype\": \"category\",\n        \"num_unique_values\": 210,\n        \"samples\": [\n          \"3 (121).jpg\",\n          \"3 (151).jpg\",\n          \"3 (111).jpg\"\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"width\",\n      \"properties\": {\n        \"dtype\": \"number\",\n        \"std\": 0,\n        \"min\": 600,\n        \"max\": 600,\n        \"num_unique_values\": 1,\n        \"samples\": [\n          600\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"height\",\n      \"properties\": {\n        \"dtype\": \"number\",\n        \"std\": 0,\n        \"min\": 337,\n        \"max\": 337,\n        \"num_unique_values\": 1,\n        \"samples\": [\n          337\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"class\",\n      \"properties\": {\n        \"dtype\": \"number\",\n        \"std\": 0,\n        \"min\": 0,\n        \"max\": 2,\n        \"num_unique_values\": 3,\n        \"samples\": [\n          0\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"xmin\",\n      \"properties\": {\n        \"dtype\": \"number\",\n        \"std\": 113.25559563317836,\n        \"min\": 87.40046838407494,\n        \"max\": 488.5011709601873,\n        \"num_unique_values\": 267,\n        \"samples\": [\n          242.03747072599532\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"ymin\",\n      \"properties\": {\n        \"dtype\": \"number\",\n        \"std\": 97.1660556798244,\n        \"min\": 0.0,\n        \"max\": 286.63956249999995,\n        \"num_unique_values\": 229,\n        \"samples\": [\n          57.92187499999999\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"xmax\",\n      \"properties\": {\n        \"dtype\": \"number\",\n        \"std\": 110.97653233844972,\n        \"min\": 99.82903981264637,\n        \"max\": 512.3185011709602,\n        \"num_unique_values\": 274,\n        \"samples\": [\n          194.8454332552693\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"ymax\",\n      \"properties\": {\n        \"dtype\": \"number\",\n        \"std\": 62.15410317148637,\n        \"min\": 94.21958333333332,\n        \"max\": 337.0,\n        \"num_unique_values\": 256,\n        \"samples\": [\n          203.26014583333333\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    }\n  ]\n}"
            }
          },
          "metadata": {},
          "execution_count": 59
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "# Сохраняем обновленный DataFrame в новый CSV-файл\n",
        "updated_labels_df.to_csv(\"new_annotations.csv\", index=False)\n",
        "\n",
        "print(\"Процесс завершен. Изображения и аннотации обновлены и сохранены.\")"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "qHlBfI2Q0GpK",
        "outputId": "a3ac58f5-0085-4de7-949a-d5839801ff86"
      },
      "execution_count": null,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Процесс завершен. Изображения и аннотации обновлены и сохранены.\n"
          ]
        }
      ]
    }
  ]
}