Cropping_Plates_from_Detection_YOLOv3_for_OCR.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "/Users/mehulgupta/Desktop\n"
     ]
    }
   ],
   "source": [
    "%cd\n",
    "%cd desktop\n",
    "import pandas as pd\n",
    "import json\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 32,
   "metadata": {},
   "outputs": [
    {
     "ename": "FileNotFoundError",
     "evalue": "[Errno 2] No such file or directory: 'result_plates.json'",
     "output_type": "error",
     "traceback": [
      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
      "\u001b[0;31mFileNotFoundError\u001b[0m                         Traceback (most recent call last)",
      "\u001b[0;32m<ipython-input-32-392e62f57a48>\u001b[0m in \u001b[0;36m<module>\u001b[0;34m\u001b[0m\n\u001b[0;32m----> 1\u001b[0;31m \u001b[0;32mwith\u001b[0m \u001b[0mopen\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m'result_plates.json'\u001b[0m\u001b[0;34m)\u001b[0m \u001b[0;32mas\u001b[0m \u001b[0mfile\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m      2\u001b[0m     \u001b[0mdata\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mjson\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mload\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mfile\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m      3\u001b[0m \u001b[0mdf\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mpd\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mDataFrame\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mdata\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;31mFileNotFoundError\u001b[0m: [Errno 2] No such file or directory: 'result_plates.json'"
     ]
    }
   ],
   "source": [
    "#loading results of YOLOv3 Detection\n",
    "with open('result_plates.json') as file:\n",
    "    data = json.load(file)\n",
    "df = pd.DataFrame(data)\n",
    "df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 33,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "/Users/mehulgupta/miniconda3/lib/python3.7/site-packages/ipykernel_launcher.py:7: SettingWithCopyWarning: \n",
      "A value is trying to be set on a copy of a slice from a DataFrame.\n",
      "Try using .loc[row_indexer,col_indexer] = value instead\n",
      "\n",
      "See the caveats in the documentation: http://pandas.pydata.org/pandas-docs/stable/user_guide/indexing.html#returning-a-view-versus-a-copy\n",
      "  import sys\n"
     ]
    }
   ],
   "source": [
    "med=[]\n",
    "def check(x):\n",
    "    for y in x:\n",
    "        if y['name']==\"NumberPlate\":\n",
    "                    return True\n",
    "    return False\n",
    "df['plates']=df['objects'].transform(lambda temp:check(temp))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 34,
   "metadata": {},
   "outputs": [],
   "source": [
    "df=df[df['plates']]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 35,
   "metadata": {},
   "outputs": [],
   "source": [
    "df['filename']=df['filename'].transform(lambda f:f.split('/')[-1])\n",
    "in_book=list(df['filename'])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 36,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "/Users/mehulgupta\n",
      "/Users/mehulgupta/Desktop/img2\n"
     ]
    }
   ],
   "source": [
    "%cd \n",
    "%cd desktop/img2\n",
    "import subprocess\n",
    "proc=subprocess.Popen('ls', shell=True, stdout=subprocess.PIPE, )\n",
    "output=proc.communicate()[0]\n",
    "output=output.decode('utf-8').split('\\n')\n",
    "output=list(output)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 37,
   "metadata": {},
   "outputs": [],
   "source": [
    "p=set(output)-set(in_book)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 38,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "usage: rm [-f | -i] [-dPRrvW] file ...\r\n",
      "       unlink file\r\n"
     ]
    }
   ],
   "source": [
    "for x in p:\n",
    "    %rm $x"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 40,
   "metadata": {},
   "outputs": [],
   "source": [
    "med=[]\n",
    "def check2(x):\n",
    "    for y in x:\n",
    "        if y['name']==\"NumberPlate\":\n",
    "                    return y\n",
    "    return False\n",
    "df['plate_data']=df['objects'].transform(lambda temp:check2(temp))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 42,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<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>frame_id</th>\n",
       "      <th>filename</th>\n",
       "      <th>objects</th>\n",
       "      <th>plates</th>\n",
       "      <th>plate_data</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>1</td>\n",
       "      <td>0.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>2</td>\n",
       "      <td>1.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>3</td>\n",
       "      <td>10.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>4</td>\n",
       "      <td>100.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>6</td>\n",
       "      <td>102.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>7</td>\n",
       "      <td>103.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>8</td>\n",
       "      <td>104.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>9</td>\n",
       "      <td>105.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>10</td>\n",
       "      <td>106.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>11</td>\n",
       "      <td>107.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>12</td>\n",
       "      <td>108.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>13</td>\n",
       "      <td>109.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>14</td>\n",
       "      <td>11.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>15</td>\n",
       "      <td>110.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>16</td>\n",
       "      <td>111.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>17</td>\n",
       "      <td>112.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>18</td>\n",
       "      <td>113.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>19</td>\n",
       "      <td>114.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>20</td>\n",
       "      <td>115.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>21</td>\n",
       "      <td>116.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>22</td>\n",
       "      <td>117.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>23</td>\n",
       "      <td>118.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23</th>\n",
       "      <td>24</td>\n",
       "      <td>119.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>25</td>\n",
       "      <td>12.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>26</td>\n",
       "      <td>120.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>27</td>\n",
       "      <td>121.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>28</td>\n",
       "      <td>122.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>28</th>\n",
       "      <td>29</td>\n",
       "      <td>123.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>29</th>\n",
       "      <td>30</td>\n",
       "      <td>124.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>30</th>\n",
       "      <td>31</td>\n",
       "      <td>125.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>112</th>\n",
       "      <td>113</td>\n",
       "      <td>72.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>113</th>\n",
       "      <td>114</td>\n",
       "      <td>73.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>114</th>\n",
       "      <td>115</td>\n",
       "      <td>74.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>115</th>\n",
       "      <td>116</td>\n",
       "      <td>75.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>116</th>\n",
       "      <td>117</td>\n",
       "      <td>76.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>117</th>\n",
       "      <td>118</td>\n",
       "      <td>77.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>118</th>\n",
       "      <td>119</td>\n",
       "      <td>78.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>119</th>\n",
       "      <td>120</td>\n",
       "      <td>79.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>120</th>\n",
       "      <td>121</td>\n",
       "      <td>8.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>121</th>\n",
       "      <td>122</td>\n",
       "      <td>80.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>122</th>\n",
       "      <td>123</td>\n",
       "      <td>81.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>123</th>\n",
       "      <td>124</td>\n",
       "      <td>82.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>124</th>\n",
       "      <td>125</td>\n",
       "      <td>83.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>125</th>\n",
       "      <td>126</td>\n",
       "      <td>84.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>126</th>\n",
       "      <td>127</td>\n",
       "      <td>85.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>127</th>\n",
       "      <td>128</td>\n",
       "      <td>86.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>128</th>\n",
       "      <td>129</td>\n",
       "      <td>87.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>129</th>\n",
       "      <td>130</td>\n",
       "      <td>88.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>130</th>\n",
       "      <td>131</td>\n",
       "      <td>89.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>131</th>\n",
       "      <td>132</td>\n",
       "      <td>9.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>132</th>\n",
       "      <td>133</td>\n",
       "      <td>90.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>133</th>\n",
       "      <td>134</td>\n",
       "      <td>91.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>134</th>\n",
       "      <td>135</td>\n",
       "      <td>92.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>135</th>\n",
       "      <td>136</td>\n",
       "      <td>93.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>136</th>\n",
       "      <td>137</td>\n",
       "      <td>94.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>137</th>\n",
       "      <td>138</td>\n",
       "      <td>95.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>138</th>\n",
       "      <td>139</td>\n",
       "      <td>96.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>139</th>\n",
       "      <td>140</td>\n",
       "      <td>97.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>140</th>\n",
       "      <td>141</td>\n",
       "      <td>98.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>141</th>\n",
       "      <td>142</td>\n",
       "      <td>99.jpg</td>\n",
       "      <td>[{'class_id': 0, 'name': 'NumberPlate', 'relat...</td>\n",
       "      <td>True</td>\n",
       "      <td>{'class_id': 0, 'name': 'NumberPlate', 'relati...</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>140 rows × 5 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "     frame_id filename                                            objects  \\\n",
       "0           1    0.jpg  [{'class_id': 0, 'name': 'NumberPlate', 'relat...   \n",
       "1           2    1.jpg  [{'class_id': 0, 'name': 'NumberPlate', 'relat...   \n",
       "2           3   10.jpg  [{'class_id': 0, 'name': 'NumberPlate', 'relat...   \n",
       "3           4  100.jpg  [{'class_id': 0, 'name': 'NumberPlate', 'relat...   \n",
       "5           6  102.jpg  [{'class_id': 0, 'name': 'NumberPlate', 'relat...   \n",
       "..        ...      ...                                                ...   \n",
       "137       138   95.jpg  [{'class_id': 0, 'name': 'NumberPlate', 'relat...   \n",
       "138       139   96.jpg  [{'class_id': 0, 'name': 'NumberPlate', 'relat...   \n",
       "139       140   97.jpg  [{'class_id': 0, 'name': 'NumberPlate', 'relat...   \n",
       "140       141   98.jpg  [{'class_id': 0, 'name': 'NumberPlate', 'relat...   \n",
       "141       142   99.jpg  [{'class_id': 0, 'name': 'NumberPlate', 'relat...   \n",
       "\n",
       "     plates                                         plate_data  \n",
       "0      True  {'class_id': 0, 'name': 'NumberPlate', 'relati...  \n",
       "1      True  {'class_id': 0, 'name': 'NumberPlate', 'relati...  \n",
       "2      True  {'class_id': 0, 'name': 'NumberPlate', 'relati...  \n",
       "3      True  {'class_id': 0, 'name': 'NumberPlate', 'relati...  \n",
       "5      True  {'class_id': 0, 'name': 'NumberPlate', 'relati...  \n",
       "..      ...                                                ...  \n",
       "137    True  {'class_id': 0, 'name': 'NumberPlate', 'relati...  \n",
       "138    True  {'class_id': 0, 'name': 'NumberPlate', 'relati...  \n",
       "139    True  {'class_id': 0, 'name': 'NumberPlate', 'relati...  \n",
       "140    True  {'class_id': 0, 'name': 'NumberPlate', 'relati...  \n",
       "141    True  {'class_id': 0, 'name': 'NumberPlate', 'relati...  \n",
       "\n",
       "[140 rows x 5 columns]"
      ]
     },
     "execution_count": 42,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 44,
   "metadata": {},
   "outputs": [],
   "source": [
    "df=df.drop(['frame_id','objects','plates'],axis=1)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 47,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'class_id': 0,\n",
       " 'name': 'NumberPlate',\n",
       " 'relative_coordinates': {'center_x': 0.49581,\n",
       "  'center_y': 0.373569,\n",
       "  'width': 0.090576,\n",
       "  'height': 0.024831},\n",
       " 'confidence': 0.474979}"
      ]
     },
     "execution_count": 47,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df['plate_data'][0]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 49,
   "metadata": {},
   "outputs": [],
   "source": [
    "data=df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 50,
   "metadata": {},
   "outputs": [],
   "source": [
    "def details(x,flag_2):\n",
    "        co=x['relative_coordinates']\n",
    "        wid=float(co['width'])/2.0\n",
    "        hi=float(co['height'])/2.0\n",
    "        if flag_2=='y2':\n",
    "            return float(co['center_y'])+hi\n",
    "        if flag_2=='x2':\n",
    "            return float(co['center_x'])+wid\n",
    "        if flag_2=='x1':\n",
    "            return float(co['center_x'])-wid\n",
    "        if flag_2=='y1':\n",
    "            return float(co['center_y'])-hi\n",
    "data['x1']=data['plate_data'].transform(lambda f:details(f,'x1'))    \n",
    "data['y1']=data['plate_data'].transform(lambda f:details(f,'y1'))  \n",
    "data['x2']=data['plate_data'].transform(lambda f:details(f,'x2'))  \n",
    "data['y2']=data['plate_data'].transform(lambda f:details(f,'y2'))  "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 56,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "0\n",
      "1\n",
      "2\n",
      "3\n",
      "4\n",
      "5\n",
      "6\n",
      "7\n",
      "8\n",
      "9\n",
      "10\n",
      "11\n",
      "12\n",
      "13\n",
      "14\n",
      "15\n",
      "16\n",
      "17\n",
      "18\n",
      "19\n",
      "20\n",
      "21\n",
      "22\n",
      "23\n",
      "24\n",
      "25\n",
      "26\n",
      "27\n",
      "28\n",
      "29\n",
      "30\n",
      "31\n",
      "32\n",
      "33\n",
      "34\n",
      "35\n",
      "36\n",
      "37\n",
      "38\n",
      "39\n",
      "40\n",
      "41\n",
      "42\n",
      "43\n",
      "44\n",
      "45\n",
      "46\n",
      "47\n",
      "48\n",
      "49\n",
      "50\n",
      "51\n",
      "52\n",
      "53\n",
      "54\n",
      "55\n",
      "56\n",
      "57\n",
      "58\n",
      "59\n",
      "60\n",
      "61\n",
      "62\n",
      "63\n",
      "64\n",
      "65\n",
      "66\n",
      "67\n",
      "68\n",
      "69\n",
      "70\n",
      "71\n",
      "72\n",
      "73\n",
      "74\n",
      "75\n",
      "76\n",
      "77\n",
      "78\n",
      "79\n",
      "80\n",
      "81\n",
      "82\n",
      "83\n",
      "84\n",
      "85\n",
      "86\n",
      "87\n",
      "88\n",
      "89\n",
      "90\n",
      "91\n",
      "92\n",
      "93\n",
      "94\n",
      "95\n",
      "96\n",
      "97\n",
      "98\n",
      "99\n",
      "100\n",
      "101\n",
      "102\n",
      "103\n",
      "104\n",
      "105\n",
      "106\n",
      "107\n",
      "108\n",
      "109\n",
      "110\n",
      "111\n",
      "112\n",
      "113\n",
      "114\n",
      "115\n",
      "116\n",
      "117\n",
      "118\n",
      "119\n",
      "120\n",
      "121\n",
      "122\n",
      "123\n",
      "124\n",
      "125\n",
      "126\n",
      "127\n",
      "128\n",
      "129\n",
      "130\n",
      "131\n",
      "132\n",
      "133\n",
      "134\n",
      "135\n",
      "136\n",
      "137\n",
      "138\n",
      "139\n"
     ]
    }
   ],
   "source": [
    "from PIL import Image\n",
    "count=0\n",
    "for x,y in data.iterrows():\n",
    "        img = Image.open('/Users/mehulgupta/Desktop/img2/'+str(y['filename']))\n",
    "        dim=img.size\n",
    "        w,h=dim[0],dim[1]\n",
    "        x1=y['x1']*w\n",
    "        x2=y['x2']*w\n",
    "        y1=y['y1']*h\n",
    "        y2=y['y2']*h\n",
    "        area=(x1,y1,x2,y2)\n",
    "        print(count)\n",
    "        cropped_img = img.crop(area)\n",
    "        cropped_img=cropped_img.convert('RGB')\n",
    "        cropped_img.save(\"/Users/mehulgupta/Desktop/im/\"+str(x)+'.jpg')\n",
    "        count+=1\n",
    "        "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.7.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}