Create_Annotation_Files_for_Images.ipynb

{"nbformat":4,"nbformat_minor":0,"metadata":{"colab":{"provenance":[],"authorship_tag":"ABX9TyO0LCtXan+mupNaqn5ZdxPr"},"kernelspec":{"name":"python3","display_name":"Python 3"},"language_info":{"name":"python"}},"cells":[{"cell_type":"code","execution_count":3,"metadata":{"colab":{"base_uri":"https://localhost:8080/"},"id":"aF93hKRzpAHQ","executionInfo":{"status":"ok","timestamp":1681341665513,"user_tz":-120,"elapsed":19060,"user":{"displayName":"Shreyas Korde","userId":"07415984173285713478"}},"outputId":"3dd41ce3-bf6b-4d21-8660-04aa7b297653"},"outputs":[{"output_type":"stream","name":"stdout","text":["Mounted at /content/drive\n"]}],"source":["from google.colab import drive\n","drive.mount('/content/drive')"]},{"cell_type":"code","source":["import pandas as pd\n","import os"],"metadata":{"id":"BVe94PjvpWmM","executionInfo":{"status":"ok","timestamp":1681341669111,"user_tz":-120,"elapsed":603,"user":{"displayName":"Shreyas Korde","userId":"07415984173285713478"}}},"execution_count":4,"outputs":[]},{"cell_type":"code","source":["os.chdir('/content/drive/My Drive/yolov4/Dataset_and_Annotations/csv_folder')\n","classes_data = pd.read_csv('class-descriptions-boxable.csv', header = None)\n","classes_data.head()"],"metadata":{"colab":{"base_uri":"https://localhost:8080/","height":206},"id":"Kev21keyps_E","executionInfo":{"status":"ok","timestamp":1681341671881,"user_tz":-120,"elapsed":836,"user":{"displayName":"Shreyas Korde","userId":"07415984173285713478"}},"outputId":"b529d0b8-a069-4f64-a7a2-3985607bef2e"},"execution_count":5,"outputs":[{"output_type":"execute_result","data":{"text/plain":["             0           1\n","0    /m/011k07    Tortoise\n","1  /m/011q46kg   Container\n","2    /m/012074      Magpie\n","3    /m/0120dh  Sea turtle\n","4    /m/01226z    Football"],"text/html":["\n","  <div id=\"df-00427a77-00d4-4bda-82e8-49ba3e3071e0\">\n","    <div 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>0</th>\n","      <th>1</th>\n","    </tr>\n","  </thead>\n","  <tbody>\n","    <tr>\n","      <th>0</th>\n","      <td>/m/011k07</td>\n","      <td>Tortoise</td>\n","    </tr>\n","    <tr>\n","      <th>1</th>\n","      <td>/m/011q46kg</td>\n","      <td>Container</td>\n","    </tr>\n","    <tr>\n","      <th>2</th>\n","      <td>/m/012074</td>\n","      <td>Magpie</td>\n","    </tr>\n","    <tr>\n","      <th>3</th>\n","      <td>/m/0120dh</td>\n","      <td>Sea turtle</td>\n","    </tr>\n","    <tr>\n","      <th>4</th>\n","      <td>/m/01226z</td>\n","      <td>Football</td>\n","    </tr>\n","  </tbody>\n","</table>\n","</div>\n","      <button class=\"colab-df-convert\" onclick=\"convertToInteractive('df-00427a77-00d4-4bda-82e8-49ba3e3071e0')\"\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 0 24 24\"\n","       width=\"24px\">\n","    <path d=\"M0 0h24v24H0V0z\" fill=\"none\"/>\n","    <path d=\"M18.56 5.44l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94zm-11 1L8.5 8.5l.94-2.06 2.06-.94-2.06-.94L8.5 2.5l-.94 2.06-2.06.94zm10 10l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94z\"/><path d=\"M17.41 7.96l-1.37-1.37c-.4-.4-.92-.59-1.43-.59-.52 0-1.04.2-1.43.59L10.3 9.45l-7.72 7.72c-.78.78-.78 2.05 0 2.83L4 21.41c.39.39.9.59 1.41.59.51 0 1.02-.2 1.41-.59l7.78-7.78 2.81-2.81c.8-.78.8-2.07 0-2.86zM5.41 20L4 18.59l7.72-7.72 1.47 1.35L5.41 20z\"/>\n","  </svg>\n","      </button>\n","      \n","  <style>\n","    .colab-df-container {\n","      display:flex;\n","      flex-wrap:wrap;\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","    [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-00427a77-00d4-4bda-82e8-49ba3e3071e0 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-00427a77-00d4-4bda-82e8-49ba3e3071e0');\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","  </div>\n","  "]},"metadata":{},"execution_count":5}]},{"cell_type":"code","source":["classes = ['Computer monitor', 'Mobile phone']\n","\n","class_string = []\n","for i in classes:\n","  req_classes = classes_data.loc[classes_data[1] == i]\n","  string = req_classes.iloc[0][0]\n","  print(string)\n","  class_string.append(string)\n","print(class_string)\n"],"metadata":{"colab":{"base_uri":"https://localhost:8080/"},"id":"Apf_UVRdyb-Y","executionInfo":{"status":"ok","timestamp":1681341674556,"user_tz":-120,"elapsed":8,"user":{"displayName":"Shreyas Korde","userId":"07415984173285713478"}},"outputId":"8d045d57-8acd-4701-e069-44fa57eaa363"},"execution_count":6,"outputs":[{"output_type":"stream","name":"stdout","text":["/m/02522\n","/m/050k8\n","['/m/02522', '/m/050k8']\n"]}]},{"cell_type":"code","source":["annotation_data = pd.read_csv('train-annotations-bbox.csv',usecols=['ImageID','LabelName', 'XMin', 'XMax', 'YMin', 'YMax'])\n","annotation_data.head()"],"metadata":{"colab":{"base_uri":"https://localhost:8080/","height":206},"id":"B1OBI8Da0lyU","executionInfo":{"status":"ok","timestamp":1681341701301,"user_tz":-120,"elapsed":24867,"user":{"displayName":"Shreyas Korde","userId":"07415984173285713478"}},"outputId":"2bdbbc5d-384d-458a-9455-0e4497cc8b5e"},"execution_count":7,"outputs":[{"output_type":"execute_result","data":{"text/plain":["            ImageID  LabelName      XMin      XMax      YMin      YMax\n","0  000002b66c9c498e  /m/01g317  0.012500  0.195312  0.148438  0.587500\n","1  000002b66c9c498e  /m/01g317  0.025000  0.276563  0.714063  0.948438\n","2  000002b66c9c498e  /m/01g317  0.151562  0.310937  0.198437  0.590625\n","3  000002b66c9c498e  /m/01g317  0.256250  0.429688  0.651563  0.925000\n","4  000002b66c9c498e  /m/01g317  0.257812  0.346875  0.235938  0.385938"],"text/html":["\n","  <div id=\"df-92c2947a-e497-49ad-82e1-a8e3b021f235\">\n","    <div 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>ImageID</th>\n","      <th>LabelName</th>\n","      <th>XMin</th>\n","      <th>XMax</th>\n","      <th>YMin</th>\n","      <th>YMax</th>\n","    </tr>\n","  </thead>\n","  <tbody>\n","    <tr>\n","      <th>0</th>\n","      <td>000002b66c9c498e</td>\n","      <td>/m/01g317</td>\n","      <td>0.012500</td>\n","      <td>0.195312</td>\n","      <td>0.148438</td>\n","      <td>0.587500</td>\n","    </tr>\n","    <tr>\n","      <th>1</th>\n","      <td>000002b66c9c498e</td>\n","      <td>/m/01g317</td>\n","      <td>0.025000</td>\n","      <td>0.276563</td>\n","      <td>0.714063</td>\n","      <td>0.948438</td>\n","    </tr>\n","    <tr>\n","      <th>2</th>\n","      <td>000002b66c9c498e</td>\n","      <td>/m/01g317</td>\n","      <td>0.151562</td>\n","      <td>0.310937</td>\n","      <td>0.198437</td>\n","      <td>0.590625</td>\n","    </tr>\n","    <tr>\n","      <th>3</th>\n","      <td>000002b66c9c498e</td>\n","      <td>/m/01g317</td>\n","      <td>0.256250</td>\n","      <td>0.429688</td>\n","      <td>0.651563</td>\n","      <td>0.925000</td>\n","    </tr>\n","    <tr>\n","      <th>4</th>\n","      <td>000002b66c9c498e</td>\n","      <td>/m/01g317</td>\n","      <td>0.257812</td>\n","      <td>0.346875</td>\n","      <td>0.235938</td>\n","      <td>0.385938</td>\n","    </tr>\n","  </tbody>\n","</table>\n","</div>\n","      <button class=\"colab-df-convert\" onclick=\"convertToInteractive('df-92c2947a-e497-49ad-82e1-a8e3b021f235')\"\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 0 24 24\"\n","       width=\"24px\">\n","    <path d=\"M0 0h24v24H0V0z\" fill=\"none\"/>\n","    <path d=\"M18.56 5.44l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94zm-11 1L8.5 8.5l.94-2.06 2.06-.94-2.06-.94L8.5 2.5l-.94 2.06-2.06.94zm10 10l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94z\"/><path d=\"M17.41 7.96l-1.37-1.37c-.4-.4-.92-.59-1.43-.59-.52 0-1.04.2-1.43.59L10.3 9.45l-7.72 7.72c-.78.78-.78 2.05 0 2.83L4 21.41c.39.39.9.59 1.41.59.51 0 1.02-.2 1.41-.59l7.78-7.78 2.81-2.81c.8-.78.8-2.07 0-2.86zM5.41 20L4 18.59l7.72-7.72 1.47 1.35L5.41 20z\"/>\n","  </svg>\n","      </button>\n","      \n","  <style>\n","    .colab-df-container {\n","      display:flex;\n","      flex-wrap:wrap;\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","    [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-92c2947a-e497-49ad-82e1-a8e3b021f235 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-92c2947a-e497-49ad-82e1-a8e3b021f235');\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","  </div>\n","  "]},"metadata":{},"execution_count":7}]},{"cell_type":"code","source":["filtered_class_data = annotation_data.loc[annotation_data['LabelName'].isin(class_string)].copy()\n","filtered_class_data.head()"],"metadata":{"colab":{"base_uri":"https://localhost:8080/","height":206},"id":"gcJTRCo72xhU","executionInfo":{"status":"ok","timestamp":1681341703907,"user_tz":-120,"elapsed":315,"user":{"displayName":"Shreyas Korde","userId":"07415984173285713478"}},"outputId":"83ad85c4-f2d3-42bd-d084-d004fd2151cc"},"execution_count":8,"outputs":[{"output_type":"execute_result","data":{"text/plain":["               ImageID LabelName      XMin      XMax      YMin      YMax\n","593   000060e3121c7305  /m/02522  0.428125  0.510625  0.540769  0.618557\n","1149  0000b9115cdf1e54  /m/050k8  0.095703  0.707031  0.497396  0.949219\n","2487  00016982f6086d39  /m/050k8  0.000000  0.997222  0.000000  0.998438\n","3963  0002a1a755d730ca  /m/050k8  0.178125  0.409375  0.339583  0.695833\n","5408  000384bb6da4764b  /m/050k8  0.695545  0.985148  0.434375  0.803750"],"text/html":["\n","  <div id=\"df-73a0ed0d-f3ab-49fc-b11b-cfaf91c2d18e\">\n","    <div 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>ImageID</th>\n","      <th>LabelName</th>\n","      <th>XMin</th>\n","      <th>XMax</th>\n","      <th>YMin</th>\n","      <th>YMax</th>\n","    </tr>\n","  </thead>\n","  <tbody>\n","    <tr>\n","      <th>593</th>\n","      <td>000060e3121c7305</td>\n","      <td>/m/02522</td>\n","      <td>0.428125</td>\n","      <td>0.510625</td>\n","      <td>0.540769</td>\n","      <td>0.618557</td>\n","    </tr>\n","    <tr>\n","      <th>1149</th>\n","      <td>0000b9115cdf1e54</td>\n","      <td>/m/050k8</td>\n","      <td>0.095703</td>\n","      <td>0.707031</td>\n","      <td>0.497396</td>\n","      <td>0.949219</td>\n","    </tr>\n","    <tr>\n","      <th>2487</th>\n","      <td>00016982f6086d39</td>\n","      <td>/m/050k8</td>\n","      <td>0.000000</td>\n","      <td>0.997222</td>\n","      <td>0.000000</td>\n","      <td>0.998438</td>\n","    </tr>\n","    <tr>\n","      <th>3963</th>\n","      <td>0002a1a755d730ca</td>\n","      <td>/m/050k8</td>\n","      <td>0.178125</td>\n","      <td>0.409375</td>\n","      <td>0.339583</td>\n","      <td>0.695833</td>\n","    </tr>\n","    <tr>\n","      <th>5408</th>\n","      <td>000384bb6da4764b</td>\n","      <td>/m/050k8</td>\n","      <td>0.695545</td>\n","      <td>0.985148</td>\n","      <td>0.434375</td>\n","      <td>0.803750</td>\n","    </tr>\n","  </tbody>\n","</table>\n","</div>\n","      <button class=\"colab-df-convert\" onclick=\"convertToInteractive('df-73a0ed0d-f3ab-49fc-b11b-cfaf91c2d18e')\"\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 0 24 24\"\n","       width=\"24px\">\n","    <path d=\"M0 0h24v24H0V0z\" fill=\"none\"/>\n","    <path d=\"M18.56 5.44l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94zm-11 1L8.5 8.5l.94-2.06 2.06-.94-2.06-.94L8.5 2.5l-.94 2.06-2.06.94zm10 10l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94z\"/><path d=\"M17.41 7.96l-1.37-1.37c-.4-.4-.92-.59-1.43-.59-.52 0-1.04.2-1.43.59L10.3 9.45l-7.72 7.72c-.78.78-.78 2.05 0 2.83L4 21.41c.39.39.9.59 1.41.59.51 0 1.02-.2 1.41-.59l7.78-7.78 2.81-2.81c.8-.78.8-2.07 0-2.86zM5.41 20L4 18.59l7.72-7.72 1.47 1.35L5.41 20z\"/>\n","  </svg>\n","      </button>\n","      \n","  <style>\n","    .colab-df-container {\n","      display:flex;\n","      flex-wrap:wrap;\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","    [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-73a0ed0d-f3ab-49fc-b11b-cfaf91c2d18e 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-73a0ed0d-f3ab-49fc-b11b-cfaf91c2d18e');\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","  </div>\n","  "]},"metadata":{},"execution_count":8}]},{"cell_type":"code","source":["# adding new columns\n","filtered_class_data ['classNumber'] = ''\n","filtered_class_data ['center x'] = ''\n","filtered_class_data ['center y'] = ''\n","filtered_class_data ['width'] = ''\n","filtered_class_data ['height'] = ''"],"metadata":{"id":"6DdLvO7M5j-H","executionInfo":{"status":"ok","timestamp":1681341706206,"user_tz":-120,"elapsed":266,"user":{"displayName":"Shreyas Korde","userId":"07415984173285713478"}}},"execution_count":9,"outputs":[]},{"cell_type":"code","source":["for i in range(len(class_string)):\n","  filtered_class_data.loc[filtered_class_data['LabelName'] == class_string[i], 'classNumber'] = i"],"metadata":{"id":"wIuxdM8Z56jQ","executionInfo":{"status":"ok","timestamp":1681341707712,"user_tz":-120,"elapsed":4,"user":{"displayName":"Shreyas Korde","userId":"07415984173285713478"}}},"execution_count":10,"outputs":[]},{"cell_type":"code","source":["filtered_class_data['center x'] = (filtered_class_data['XMax'] + filtered_class_data['XMin']) / 2\n","filtered_class_data['center y'] = (filtered_class_data['YMax'] + filtered_class_data['YMin']) / 2\n"],"metadata":{"id":"FJMgsKk86kny","executionInfo":{"status":"ok","timestamp":1681341708719,"user_tz":-120,"elapsed":3,"user":{"displayName":"Shreyas Korde","userId":"07415984173285713478"}}},"execution_count":11,"outputs":[]},{"cell_type":"code","source":["filtered_class_data.head()"],"metadata":{"colab":{"base_uri":"https://localhost:8080/","height":206},"id":"xXcfWoUl7QiC","executionInfo":{"status":"ok","timestamp":1681341711136,"user_tz":-120,"elapsed":17,"user":{"displayName":"Shreyas Korde","userId":"07415984173285713478"}},"outputId":"87628b77-67c3-4294-ff61-ba336f14be12"},"execution_count":12,"outputs":[{"output_type":"execute_result","data":{"text/plain":["               ImageID LabelName      XMin      XMax      YMin      YMax  \\\n","593   000060e3121c7305  /m/02522  0.428125  0.510625  0.540769  0.618557   \n","1149  0000b9115cdf1e54  /m/050k8  0.095703  0.707031  0.497396  0.949219   \n","2487  00016982f6086d39  /m/050k8  0.000000  0.997222  0.000000  0.998438   \n","3963  0002a1a755d730ca  /m/050k8  0.178125  0.409375  0.339583  0.695833   \n","5408  000384bb6da4764b  /m/050k8  0.695545  0.985148  0.434375  0.803750   \n","\n","     classNumber  center x  center y width height  \n","593            0  0.469375  0.579663               \n","1149           1  0.401367  0.723307               \n","2487           1  0.498611  0.499219               \n","3963           1  0.293750  0.517708               \n","5408           1  0.840346  0.619062               "],"text/html":["\n","  <div id=\"df-c88ad1f6-30b7-43a8-a690-fe3fbcc6559f\">\n","    <div 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>ImageID</th>\n","      <th>LabelName</th>\n","      <th>XMin</th>\n","      <th>XMax</th>\n","      <th>YMin</th>\n","      <th>YMax</th>\n","      <th>classNumber</th>\n","      <th>center x</th>\n","      <th>center y</th>\n","      <th>width</th>\n","      <th>height</th>\n","    </tr>\n","  </thead>\n","  <tbody>\n","    <tr>\n","      <th>593</th>\n","      <td>000060e3121c7305</td>\n","      <td>/m/02522</td>\n","      <td>0.428125</td>\n","      <td>0.510625</td>\n","      <td>0.540769</td>\n","      <td>0.618557</td>\n","      <td>0</td>\n","      <td>0.469375</td>\n","      <td>0.579663</td>\n","      <td></td>\n","      <td></td>\n","    </tr>\n","    <tr>\n","      <th>1149</th>\n","      <td>0000b9115cdf1e54</td>\n","      <td>/m/050k8</td>\n","      <td>0.095703</td>\n","      <td>0.707031</td>\n","      <td>0.497396</td>\n","      <td>0.949219</td>\n","      <td>1</td>\n","      <td>0.401367</td>\n","      <td>0.723307</td>\n","      <td></td>\n","      <td></td>\n","    </tr>\n","    <tr>\n","      <th>2487</th>\n","      <td>00016982f6086d39</td>\n","      <td>/m/050k8</td>\n","      <td>0.000000</td>\n","      <td>0.997222</td>\n","      <td>0.000000</td>\n","      <td>0.998438</td>\n","      <td>1</td>\n","      <td>0.498611</td>\n","      <td>0.499219</td>\n","      <td></td>\n","      <td></td>\n","    </tr>\n","    <tr>\n","      <th>3963</th>\n","      <td>0002a1a755d730ca</td>\n","      <td>/m/050k8</td>\n","      <td>0.178125</td>\n","      <td>0.409375</td>\n","      <td>0.339583</td>\n","      <td>0.695833</td>\n","      <td>1</td>\n","      <td>0.293750</td>\n","      <td>0.517708</td>\n","      <td></td>\n","      <td></td>\n","    </tr>\n","    <tr>\n","      <th>5408</th>\n","      <td>000384bb6da4764b</td>\n","      <td>/m/050k8</td>\n","      <td>0.695545</td>\n","      <td>0.985148</td>\n","      <td>0.434375</td>\n","      <td>0.803750</td>\n","      <td>1</td>\n","      <td>0.840346</td>\n","      <td>0.619062</td>\n","      <td></td>\n","      <td></td>\n","    </tr>\n","  </tbody>\n","</table>\n","</div>\n","      <button class=\"colab-df-convert\" onclick=\"convertToInteractive('df-c88ad1f6-30b7-43a8-a690-fe3fbcc6559f')\"\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 0 24 24\"\n","       width=\"24px\">\n","    <path d=\"M0 0h24v24H0V0z\" fill=\"none\"/>\n","    <path d=\"M18.56 5.44l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94zm-11 1L8.5 8.5l.94-2.06 2.06-.94-2.06-.94L8.5 2.5l-.94 2.06-2.06.94zm10 10l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94z\"/><path d=\"M17.41 7.96l-1.37-1.37c-.4-.4-.92-.59-1.43-.59-.52 0-1.04.2-1.43.59L10.3 9.45l-7.72 7.72c-.78.78-.78 2.05 0 2.83L4 21.41c.39.39.9.59 1.41.59.51 0 1.02-.2 1.41-.59l7.78-7.78 2.81-2.81c.8-.78.8-2.07 0-2.86zM5.41 20L4 18.59l7.72-7.72 1.47 1.35L5.41 20z\"/>\n","  </svg>\n","      </button>\n","      \n","  <style>\n","    .colab-df-container {\n","      display:flex;\n","      flex-wrap:wrap;\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","    [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-c88ad1f6-30b7-43a8-a690-fe3fbcc6559f 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-c88ad1f6-30b7-43a8-a690-fe3fbcc6559f');\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","  </div>\n","  "]},"metadata":{},"execution_count":12}]},{"cell_type":"code","source":["filtered_class_data['width'] = filtered_class_data['XMax'] - filtered_class_data['XMin'] \n","filtered_class_data['height'] = filtered_class_data['YMax'] - filtered_class_data['YMin'] \n","filtered_class_data.head()"],"metadata":{"colab":{"base_uri":"https://localhost:8080/","height":206},"id":"U18TpnrL7UHo","executionInfo":{"status":"ok","timestamp":1681341713077,"user_tz":-120,"elapsed":15,"user":{"displayName":"Shreyas Korde","userId":"07415984173285713478"}},"outputId":"cc979d73-8a2d-4100-b17e-629dc945e7c5"},"execution_count":13,"outputs":[{"output_type":"execute_result","data":{"text/plain":["               ImageID LabelName      XMin      XMax      YMin      YMax  \\\n","593   000060e3121c7305  /m/02522  0.428125  0.510625  0.540769  0.618557   \n","1149  0000b9115cdf1e54  /m/050k8  0.095703  0.707031  0.497396  0.949219   \n","2487  00016982f6086d39  /m/050k8  0.000000  0.997222  0.000000  0.998438   \n","3963  0002a1a755d730ca  /m/050k8  0.178125  0.409375  0.339583  0.695833   \n","5408  000384bb6da4764b  /m/050k8  0.695545  0.985148  0.434375  0.803750   \n","\n","     classNumber  center x  center y     width    height  \n","593            0  0.469375  0.579663  0.082500  0.077788  \n","1149           1  0.401367  0.723307  0.611328  0.451823  \n","2487           1  0.498611  0.499219  0.997222  0.998438  \n","3963           1  0.293750  0.517708  0.231250  0.356250  \n","5408           1  0.840346  0.619062  0.289603  0.369375  "],"text/html":["\n","  <div id=\"df-5818c963-71cd-4bb5-8774-b484c34daa03\">\n","    <div 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>ImageID</th>\n","      <th>LabelName</th>\n","      <th>XMin</th>\n","      <th>XMax</th>\n","      <th>YMin</th>\n","      <th>YMax</th>\n","      <th>classNumber</th>\n","      <th>center x</th>\n","      <th>center y</th>\n","      <th>width</th>\n","      <th>height</th>\n","    </tr>\n","  </thead>\n","  <tbody>\n","    <tr>\n","      <th>593</th>\n","      <td>000060e3121c7305</td>\n","      <td>/m/02522</td>\n","      <td>0.428125</td>\n","      <td>0.510625</td>\n","      <td>0.540769</td>\n","      <td>0.618557</td>\n","      <td>0</td>\n","      <td>0.469375</td>\n","      <td>0.579663</td>\n","      <td>0.082500</td>\n","      <td>0.077788</td>\n","    </tr>\n","    <tr>\n","      <th>1149</th>\n","      <td>0000b9115cdf1e54</td>\n","      <td>/m/050k8</td>\n","      <td>0.095703</td>\n","      <td>0.707031</td>\n","      <td>0.497396</td>\n","      <td>0.949219</td>\n","      <td>1</td>\n","      <td>0.401367</td>\n","      <td>0.723307</td>\n","      <td>0.611328</td>\n","      <td>0.451823</td>\n","    </tr>\n","    <tr>\n","      <th>2487</th>\n","      <td>00016982f6086d39</td>\n","      <td>/m/050k8</td>\n","      <td>0.000000</td>\n","      <td>0.997222</td>\n","      <td>0.000000</td>\n","      <td>0.998438</td>\n","      <td>1</td>\n","      <td>0.498611</td>\n","      <td>0.499219</td>\n","      <td>0.997222</td>\n","      <td>0.998438</td>\n","    </tr>\n","    <tr>\n","      <th>3963</th>\n","      <td>0002a1a755d730ca</td>\n","      <td>/m/050k8</td>\n","      <td>0.178125</td>\n","      <td>0.409375</td>\n","      <td>0.339583</td>\n","      <td>0.695833</td>\n","      <td>1</td>\n","      <td>0.293750</td>\n","      <td>0.517708</td>\n","      <td>0.231250</td>\n","      <td>0.356250</td>\n","    </tr>\n","    <tr>\n","      <th>5408</th>\n","      <td>000384bb6da4764b</td>\n","      <td>/m/050k8</td>\n","      <td>0.695545</td>\n","      <td>0.985148</td>\n","      <td>0.434375</td>\n","      <td>0.803750</td>\n","      <td>1</td>\n","      <td>0.840346</td>\n","      <td>0.619062</td>\n","      <td>0.289603</td>\n","      <td>0.369375</td>\n","    </tr>\n","  </tbody>\n","</table>\n","</div>\n","      <button class=\"colab-df-convert\" onclick=\"convertToInteractive('df-5818c963-71cd-4bb5-8774-b484c34daa03')\"\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 0 24 24\"\n","       width=\"24px\">\n","    <path d=\"M0 0h24v24H0V0z\" fill=\"none\"/>\n","    <path d=\"M18.56 5.44l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94zm-11 1L8.5 8.5l.94-2.06 2.06-.94-2.06-.94L8.5 2.5l-.94 2.06-2.06.94zm10 10l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94z\"/><path d=\"M17.41 7.96l-1.37-1.37c-.4-.4-.92-.59-1.43-.59-.52 0-1.04.2-1.43.59L10.3 9.45l-7.72 7.72c-.78.78-.78 2.05 0 2.83L4 21.41c.39.39.9.59 1.41.59.51 0 1.02-.2 1.41-.59l7.78-7.78 2.81-2.81c.8-.78.8-2.07 0-2.86zM5.41 20L4 18.59l7.72-7.72 1.47 1.35L5.41 20z\"/>\n","  </svg>\n","      </button>\n","      \n","  <style>\n","    .colab-df-container {\n","      display:flex;\n","      flex-wrap:wrap;\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","    [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-5818c963-71cd-4bb5-8774-b484c34daa03 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-5818c963-71cd-4bb5-8774-b484c34daa03');\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","  </div>\n","  "]},"metadata":{},"execution_count":13}]},{"cell_type":"code","source":["YOLO_values = filtered_class_data.loc[:, ['ImageID', 'classNumber', 'center x', 'center y', 'width', 'height']].copy()\n","YOLO_values.head()\n","len(YOLO_values)"],"metadata":{"colab":{"base_uri":"https://localhost:8080/"},"id":"LLv_b5GJ74sK","executionInfo":{"status":"ok","timestamp":1681341714508,"user_tz":-120,"elapsed":7,"user":{"displayName":"Shreyas Korde","userId":"07415984173285713478"}},"outputId":"ca37a84a-3c06-4e7c-d430-71eb7db8d93d"},"execution_count":14,"outputs":[{"output_type":"execute_result","data":{"text/plain":["12477"]},"metadata":{},"execution_count":14}]},{"cell_type":"code","source":["image_path = '/content/drive/My Drive/yolov4/Dataset_and_Annotations/Dataset/train/Computer monitor_Mobile phone'\n","os.chdir(image_path)\n","\n","for current_dir, dirs, files in os.walk('.'):\n","  for f in files:\n","    if f.endswith('.jpg'):\n","      image_title =f[:-4]\n","      YOLO_file = YOLO_values.loc[YOLO_values['ImageID'] == image_title]\n","\n","      df =  YOLO_file.loc[:, ['classNumber', 'center x', 'center y', 'width', 'height']].copy()\n","      save_path = image_path + '/' + image_title + '.txt'\n","\n","      df.to_csv(save_path, header=False, index=False, sep = ' ')"],"metadata":{"id":"Xv_h4hFC8Wqv","executionInfo":{"status":"ok","timestamp":1681341773456,"user_tz":-120,"elapsed":29996,"user":{"displayName":"Shreyas Korde","userId":"07415984173285713478"}}},"execution_count":17,"outputs":[]},{"cell_type":"code","source":[],"metadata":{"id":"0i_PflnrxMt8"},"execution_count":null,"outputs":[]}]}