notebook/prepare_lgb_test.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "import numpy as np\n",
    "from ast import literal_eval\n",
    "\n",
    "IMG_WIDTH = 1914\n",
    "IMG_HEIGHT = 1052"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "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>guid/image</th>\n",
       "      <th>cocov1-can1-label</th>\n",
       "      <th>cocov1-can1-score</th>\n",
       "      <th>cocov1-can1-ymin</th>\n",
       "      <th>cocov1-can1-xmin</th>\n",
       "      <th>cocov1-can1-ymax</th>\n",
       "      <th>cocov1-can1-xmax</th>\n",
       "      <th>cocov1-can1-box-width</th>\n",
       "      <th>cocov1-can1-box-height</th>\n",
       "      <th>cocov1-can1-box-ratio</th>\n",
       "      <th>cocov1-can1-box-area</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0729ab2d-7fb0-4799-975f-c38fd350bf9c/0090</td>\n",
       "      <td>2</td>\n",
       "      <td>0.081078</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>1699.343391</td>\n",
       "      <td>954.240316</td>\n",
       "      <td>1914.000000</td>\n",
       "      <td>214.656609</td>\n",
       "      <td>954.240316</td>\n",
       "      <td>4.445427</td>\n",
       "      <td>204833.990316</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>72e86196-dd16-4d1a-8285-26057e0aea6a/0017</td>\n",
       "      <td>1</td>\n",
       "      <td>0.048925</td>\n",
       "      <td>406.972815</td>\n",
       "      <td>966.574098</td>\n",
       "      <td>472.374932</td>\n",
       "      <td>1094.256344</td>\n",
       "      <td>127.682246</td>\n",
       "      <td>65.402118</td>\n",
       "      <td>0.512226</td>\n",
       "      <td>8350.689314</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>748e842e-ed85-45ca-b8f1-4acdcb76f3eb/0200</td>\n",
       "      <td>1</td>\n",
       "      <td>0.999940</td>\n",
       "      <td>382.589704</td>\n",
       "      <td>397.916755</td>\n",
       "      <td>449.134290</td>\n",
       "      <td>480.911778</td>\n",
       "      <td>82.995023</td>\n",
       "      <td>66.544586</td>\n",
       "      <td>0.801790</td>\n",
       "      <td>5522.869469</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>e024982a-233f-45b4-a6ec-505f5b0c2649/0058</td>\n",
       "      <td>0</td>\n",
       "      <td>0.999998</td>\n",
       "      <td>109.573384</td>\n",
       "      <td>829.275771</td>\n",
       "      <td>608.459071</td>\n",
       "      <td>1242.696730</td>\n",
       "      <td>413.420959</td>\n",
       "      <td>498.885687</td>\n",
       "      <td>1.206726</td>\n",
       "      <td>206249.799195</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>e024982a-233f-45b4-a6ec-505f5b0c2649/0004</td>\n",
       "      <td>2</td>\n",
       "      <td>0.988845</td>\n",
       "      <td>182.010195</td>\n",
       "      <td>924.366817</td>\n",
       "      <td>550.146967</td>\n",
       "      <td>1214.544054</td>\n",
       "      <td>290.177237</td>\n",
       "      <td>368.136772</td>\n",
       "      <td>1.268662</td>\n",
       "      <td>106824.911447</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                                  guid/image  cocov1-can1-label  \\\n",
       "0  0729ab2d-7fb0-4799-975f-c38fd350bf9c/0090                  2   \n",
       "1  72e86196-dd16-4d1a-8285-26057e0aea6a/0017                  1   \n",
       "2  748e842e-ed85-45ca-b8f1-4acdcb76f3eb/0200                  1   \n",
       "3  e024982a-233f-45b4-a6ec-505f5b0c2649/0058                  0   \n",
       "4  e024982a-233f-45b4-a6ec-505f5b0c2649/0004                  2   \n",
       "\n",
       "   cocov1-can1-score  cocov1-can1-ymin  cocov1-can1-xmin  cocov1-can1-ymax  \\\n",
       "0           0.081078          0.000000       1699.343391        954.240316   \n",
       "1           0.048925        406.972815        966.574098        472.374932   \n",
       "2           0.999940        382.589704        397.916755        449.134290   \n",
       "3           0.999998        109.573384        829.275771        608.459071   \n",
       "4           0.988845        182.010195        924.366817        550.146967   \n",
       "\n",
       "   cocov1-can1-xmax  cocov1-can1-box-width  cocov1-can1-box-height  \\\n",
       "0       1914.000000             214.656609              954.240316   \n",
       "1       1094.256344             127.682246               65.402118   \n",
       "2        480.911778              82.995023               66.544586   \n",
       "3       1242.696730             413.420959              498.885687   \n",
       "4       1214.544054             290.177237              368.136772   \n",
       "\n",
       "   cocov1-can1-box-ratio  cocov1-can1-box-area  \n",
       "0               4.445427         204833.990316  \n",
       "1               0.512226           8350.689314  \n",
       "2               0.801790           5522.869469  \n",
       "3               1.206726         206249.799195  \n",
       "4               1.268662         106824.911447  "
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "cocov1 = pd.read_csv('tf_features/cocov1-steps-201k-kaggle-test.tsv',sep='\\t')\n",
    "\n",
    "cocov1['cocov1-can1-label'] = cocov1['labels'].apply(lambda x : int(x[1]))\n",
    "cocov1['cocov1-can1-score'] = cocov1['scores'].apply(lambda x : literal_eval(x)[0])\n",
    "cocov1['cocov1-can1-ymin'] = cocov1['boxes'].apply(lambda x : IMG_HEIGHT*literal_eval(x)[0][0])\n",
    "cocov1['cocov1-can1-xmin'] = cocov1['boxes'].apply(lambda x : IMG_WIDTH*literal_eval(x)[0][1])\n",
    "cocov1['cocov1-can1-ymax'] = cocov1['boxes'].apply(lambda x : IMG_HEIGHT*literal_eval(x)[0][2])\n",
    "cocov1['cocov1-can1-xmax'] = cocov1['boxes'].apply(lambda x : IMG_WIDTH*literal_eval(x)[0][3])\n",
    "cocov1['cocov1-can1-box-width'] = cocov1['cocov1-can1-xmax'] - cocov1['cocov1-can1-xmin']\n",
    "cocov1['cocov1-can1-box-height'] = cocov1['cocov1-can1-ymax'] - cocov1['cocov1-can1-ymin']\n",
    "cocov1['cocov1-can1-box-ratio'] = cocov1['cocov1-can1-box-height']/cocov1['cocov1-can1-box-width']\n",
    "cocov1['cocov1-can1-box-area'] = cocov1['cocov1-can1-box-height']*cocov1['cocov1-can1-box-width']\n",
    "\n",
    "\n",
    "cocov1 = cocov1.drop(['labels','scores','boxes'],axis=1)\n",
    "cocov1.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "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>guid/image</th>\n",
       "      <th>kitti-can1-label</th>\n",
       "      <th>kitti-can1-score</th>\n",
       "      <th>kitti-can1-ymin</th>\n",
       "      <th>kitti-can1-xmin</th>\n",
       "      <th>kitti-can1-ymax</th>\n",
       "      <th>kitti-can1-xmax</th>\n",
       "      <th>kitti-can1-box-width</th>\n",
       "      <th>kitti-can1-box-height</th>\n",
       "      <th>kitti-can1-box-ratio</th>\n",
       "      <th>kitti-can1-box-area</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0729ab2d-7fb0-4799-975f-c38fd350bf9c/0090</td>\n",
       "      <td>0</td>\n",
       "      <td>0.777016</td>\n",
       "      <td>418.017857</td>\n",
       "      <td>1538.341409</td>\n",
       "      <td>481.414040</td>\n",
       "      <td>1621.456561</td>\n",
       "      <td>83.115153</td>\n",
       "      <td>63.396183</td>\n",
       "      <td>0.762751</td>\n",
       "      <td>5269.183411</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>72e86196-dd16-4d1a-8285-26057e0aea6a/0017</td>\n",
       "      <td>0</td>\n",
       "      <td>0.999581</td>\n",
       "      <td>227.323287</td>\n",
       "      <td>1514.442214</td>\n",
       "      <td>537.447509</td>\n",
       "      <td>1824.921714</td>\n",
       "      <td>310.479499</td>\n",
       "      <td>310.124221</td>\n",
       "      <td>0.998856</td>\n",
       "      <td>96287.213007</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>748e842e-ed85-45ca-b8f1-4acdcb76f3eb/0200</td>\n",
       "      <td>1</td>\n",
       "      <td>0.999993</td>\n",
       "      <td>388.680434</td>\n",
       "      <td>394.252713</td>\n",
       "      <td>443.291116</td>\n",
       "      <td>481.365430</td>\n",
       "      <td>87.112717</td>\n",
       "      <td>54.610682</td>\n",
       "      <td>0.626897</td>\n",
       "      <td>4757.284860</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>e024982a-233f-45b4-a6ec-505f5b0c2649/0058</td>\n",
       "      <td>0</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>116.014379</td>\n",
       "      <td>826.515583</td>\n",
       "      <td>608.119654</td>\n",
       "      <td>1247.116088</td>\n",
       "      <td>420.600506</td>\n",
       "      <td>492.105275</td>\n",
       "      <td>1.170006</td>\n",
       "      <td>206979.727412</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>e024982a-233f-45b4-a6ec-505f5b0c2649/0004</td>\n",
       "      <td>2</td>\n",
       "      <td>0.999974</td>\n",
       "      <td>165.650918</td>\n",
       "      <td>922.230950</td>\n",
       "      <td>561.412321</td>\n",
       "      <td>1217.553229</td>\n",
       "      <td>295.322279</td>\n",
       "      <td>395.761402</td>\n",
       "      <td>1.340100</td>\n",
       "      <td>116877.159438</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                                  guid/image  kitti-can1-label  \\\n",
       "0  0729ab2d-7fb0-4799-975f-c38fd350bf9c/0090                 0   \n",
       "1  72e86196-dd16-4d1a-8285-26057e0aea6a/0017                 0   \n",
       "2  748e842e-ed85-45ca-b8f1-4acdcb76f3eb/0200                 1   \n",
       "3  e024982a-233f-45b4-a6ec-505f5b0c2649/0058                 0   \n",
       "4  e024982a-233f-45b4-a6ec-505f5b0c2649/0004                 2   \n",
       "\n",
       "   kitti-can1-score  kitti-can1-ymin  kitti-can1-xmin  kitti-can1-ymax  \\\n",
       "0          0.777016       418.017857      1538.341409       481.414040   \n",
       "1          0.999581       227.323287      1514.442214       537.447509   \n",
       "2          0.999993       388.680434       394.252713       443.291116   \n",
       "3          1.000000       116.014379       826.515583       608.119654   \n",
       "4          0.999974       165.650918       922.230950       561.412321   \n",
       "\n",
       "   kitti-can1-xmax  kitti-can1-box-width  kitti-can1-box-height  \\\n",
       "0      1621.456561             83.115153              63.396183   \n",
       "1      1824.921714            310.479499             310.124221   \n",
       "2       481.365430             87.112717              54.610682   \n",
       "3      1247.116088            420.600506             492.105275   \n",
       "4      1217.553229            295.322279             395.761402   \n",
       "\n",
       "   kitti-can1-box-ratio  kitti-can1-box-area  \n",
       "0              0.762751          5269.183411  \n",
       "1              0.998856         96287.213007  \n",
       "2              0.626897          4757.284860  \n",
       "3              1.170006        206979.727412  \n",
       "4              1.340100        116877.159438  "
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "kitti = pd.read_csv('tf_features/kitti-steps-660k-kaggle-test.tsv',sep='\\t')\n",
    "\n",
    "kitti['kitti-can1-label'] = kitti['labels'].apply(lambda x : int(x[1]))\n",
    "kitti['kitti-can1-score'] = kitti['scores'].apply(lambda x : literal_eval(x)[0])\n",
    "kitti['kitti-can1-ymin'] = kitti['boxes'].apply(lambda x : IMG_HEIGHT*literal_eval(x)[0][0])\n",
    "kitti['kitti-can1-xmin'] = kitti['boxes'].apply(lambda x : IMG_WIDTH*literal_eval(x)[0][1])\n",
    "kitti['kitti-can1-ymax'] = kitti['boxes'].apply(lambda x : IMG_HEIGHT*literal_eval(x)[0][2])\n",
    "kitti['kitti-can1-xmax'] = kitti['boxes'].apply(lambda x : IMG_WIDTH*literal_eval(x)[0][3])\n",
    "\n",
    "kitti['kitti-can1-box-width'] = kitti['kitti-can1-xmax'] - kitti['kitti-can1-xmin']\n",
    "kitti['kitti-can1-box-height'] = kitti['kitti-can1-ymax'] - kitti['kitti-can1-ymin']\n",
    "kitti['kitti-can1-box-ratio'] = kitti['kitti-can1-box-height']/kitti['kitti-can1-box-width']\n",
    "kitti['kitti-can1-box-area'] = kitti['kitti-can1-box-height']*kitti['kitti-can1-box-width']\n",
    "\n",
    "kitti = kitti.drop(['labels','scores','boxes'],axis=1)\n",
    "kitti.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "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>guid/image</th>\n",
       "      <th>incep-can1-label</th>\n",
       "      <th>incep-can1-score</th>\n",
       "      <th>incep-can1-ymin</th>\n",
       "      <th>incep-can1-xmin</th>\n",
       "      <th>incep-can1-ymax</th>\n",
       "      <th>incep-can1-xmax</th>\n",
       "      <th>incep-can1-box-width</th>\n",
       "      <th>incep-can1-box-height</th>\n",
       "      <th>incep-can1-box-ratio</th>\n",
       "      <th>incep-can1-box-area</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0729ab2d-7fb0-4799-975f-c38fd350bf9c/0090</td>\n",
       "      <td>1</td>\n",
       "      <td>0.017475</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>566.255710</td>\n",
       "      <td>501.622401</td>\n",
       "      <td>501.622401</td>\n",
       "      <td>566.255710</td>\n",
       "      <td>1.128849</td>\n",
       "      <td>284046.548908</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>72e86196-dd16-4d1a-8285-26057e0aea6a/0017</td>\n",
       "      <td>1</td>\n",
       "      <td>0.999968</td>\n",
       "      <td>419.162144</td>\n",
       "      <td>959.193365</td>\n",
       "      <td>455.486559</td>\n",
       "      <td>1013.015124</td>\n",
       "      <td>53.821758</td>\n",
       "      <td>36.324414</td>\n",
       "      <td>0.674902</td>\n",
       "      <td>1955.043856</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>748e842e-ed85-45ca-b8f1-4acdcb76f3eb/0200</td>\n",
       "      <td>1</td>\n",
       "      <td>0.999996</td>\n",
       "      <td>388.984611</td>\n",
       "      <td>393.755196</td>\n",
       "      <td>442.966120</td>\n",
       "      <td>483.278607</td>\n",
       "      <td>89.523411</td>\n",
       "      <td>53.981509</td>\n",
       "      <td>0.602988</td>\n",
       "      <td>4832.608808</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>e024982a-233f-45b4-a6ec-505f5b0c2649/0058</td>\n",
       "      <td>0</td>\n",
       "      <td>0.999925</td>\n",
       "      <td>121.529517</td>\n",
       "      <td>825.027480</td>\n",
       "      <td>596.204812</td>\n",
       "      <td>1246.526848</td>\n",
       "      <td>421.499368</td>\n",
       "      <td>474.675295</td>\n",
       "      <td>1.126159</td>\n",
       "      <td>200075.336594</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>e024982a-233f-45b4-a6ec-505f5b0c2649/0004</td>\n",
       "      <td>2</td>\n",
       "      <td>0.995085</td>\n",
       "      <td>178.878596</td>\n",
       "      <td>919.342247</td>\n",
       "      <td>562.062876</td>\n",
       "      <td>1216.394371</td>\n",
       "      <td>297.052124</td>\n",
       "      <td>383.184280</td>\n",
       "      <td>1.289956</td>\n",
       "      <td>113825.704210</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                                  guid/image  incep-can1-label  \\\n",
       "0  0729ab2d-7fb0-4799-975f-c38fd350bf9c/0090                 1   \n",
       "1  72e86196-dd16-4d1a-8285-26057e0aea6a/0017                 1   \n",
       "2  748e842e-ed85-45ca-b8f1-4acdcb76f3eb/0200                 1   \n",
       "3  e024982a-233f-45b4-a6ec-505f5b0c2649/0058                 0   \n",
       "4  e024982a-233f-45b4-a6ec-505f5b0c2649/0004                 2   \n",
       "\n",
       "   incep-can1-score  incep-can1-ymin  incep-can1-xmin  incep-can1-ymax  \\\n",
       "0          0.017475         0.000000         0.000000       566.255710   \n",
       "1          0.999968       419.162144       959.193365       455.486559   \n",
       "2          0.999996       388.984611       393.755196       442.966120   \n",
       "3          0.999925       121.529517       825.027480       596.204812   \n",
       "4          0.995085       178.878596       919.342247       562.062876   \n",
       "\n",
       "   incep-can1-xmax  incep-can1-box-width  incep-can1-box-height  \\\n",
       "0       501.622401            501.622401             566.255710   \n",
       "1      1013.015124             53.821758              36.324414   \n",
       "2       483.278607             89.523411              53.981509   \n",
       "3      1246.526848            421.499368             474.675295   \n",
       "4      1216.394371            297.052124             383.184280   \n",
       "\n",
       "   incep-can1-box-ratio  incep-can1-box-area  \n",
       "0              1.128849        284046.548908  \n",
       "1              0.674902          1955.043856  \n",
       "2              0.602988          4832.608808  \n",
       "3              1.126159        200075.336594  \n",
       "4              1.289956        113825.704210  "
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "incep = pd.read_csv('tf_features/incep-steps-200k-kaggle-test.tsv',sep='\\t')\n",
    "\n",
    "incep['incep-can1-label'] = incep['labels'].apply(lambda x : int(x[1]))\n",
    "incep['incep-can1-score'] = incep['scores'].apply(lambda x : literal_eval(x)[0])\n",
    "incep['incep-can1-ymin'] = incep['boxes'].apply(lambda x : IMG_HEIGHT*literal_eval(x)[0][0])\n",
    "incep['incep-can1-xmin'] = incep['boxes'].apply(lambda x : IMG_WIDTH*literal_eval(x)[0][1])\n",
    "incep['incep-can1-ymax'] = incep['boxes'].apply(lambda x : IMG_HEIGHT*literal_eval(x)[0][2])\n",
    "incep['incep-can1-xmax'] = incep['boxes'].apply(lambda x : IMG_WIDTH*literal_eval(x)[0][3])\n",
    "\n",
    "incep['incep-can1-box-width'] = incep['incep-can1-xmax'] - incep['incep-can1-xmin']\n",
    "incep['incep-can1-box-height'] = incep['incep-can1-ymax'] - incep['incep-can1-ymin']\n",
    "incep['incep-can1-box-ratio'] = incep['incep-can1-box-height']/incep['incep-can1-box-width']\n",
    "incep['incep-can1-box-area'] = incep['incep-can1-box-height']*incep['incep-can1-box-width']\n",
    "\n",
    "incep = incep.drop(['labels','scores','boxes'],axis=1)\n",
    "incep.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "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>guid/image</th>\n",
       "      <th>cocov1-can1-label</th>\n",
       "      <th>cocov1-can1-score</th>\n",
       "      <th>cocov1-can1-ymin</th>\n",
       "      <th>cocov1-can1-xmin</th>\n",
       "      <th>cocov1-can1-ymax</th>\n",
       "      <th>cocov1-can1-xmax</th>\n",
       "      <th>cocov1-can1-box-width</th>\n",
       "      <th>cocov1-can1-box-height</th>\n",
       "      <th>cocov1-can1-box-ratio</th>\n",
       "      <th>...</th>\n",
       "      <th>incep-can1-label</th>\n",
       "      <th>incep-can1-score</th>\n",
       "      <th>incep-can1-ymin</th>\n",
       "      <th>incep-can1-xmin</th>\n",
       "      <th>incep-can1-ymax</th>\n",
       "      <th>incep-can1-xmax</th>\n",
       "      <th>incep-can1-box-width</th>\n",
       "      <th>incep-can1-box-height</th>\n",
       "      <th>incep-can1-box-ratio</th>\n",
       "      <th>incep-can1-box-area</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0410e7f5-4bfd-4d35-b623-9f8d6b87c179/0000</td>\n",
       "      <td>0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>105.870206</td>\n",
       "      <td>778.004116</td>\n",
       "      <td>599.913759</td>\n",
       "      <td>1198.261631</td>\n",
       "      <td>420.257514</td>\n",
       "      <td>494.043552</td>\n",
       "      <td>1.175573</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0.999922</td>\n",
       "      <td>114.727974</td>\n",
       "      <td>778.541677</td>\n",
       "      <td>597.486985</td>\n",
       "      <td>1199.429844</td>\n",
       "      <td>420.888167</td>\n",
       "      <td>482.759011</td>\n",
       "      <td>1.147001</td>\n",
       "      <td>203187.555194</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>0410e7f5-4bfd-4d35-b623-9f8d6b87c179/0001</td>\n",
       "      <td>0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>107.833644</td>\n",
       "      <td>768.235507</td>\n",
       "      <td>598.237929</td>\n",
       "      <td>1182.570957</td>\n",
       "      <td>414.335451</td>\n",
       "      <td>490.404285</td>\n",
       "      <td>1.183592</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0.999919</td>\n",
       "      <td>118.921772</td>\n",
       "      <td>768.581806</td>\n",
       "      <td>588.000484</td>\n",
       "      <td>1186.015816</td>\n",
       "      <td>417.434010</td>\n",
       "      <td>469.078712</td>\n",
       "      <td>1.123719</td>\n",
       "      <td>195809.407618</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>0410e7f5-4bfd-4d35-b623-9f8d6b87c179/0002</td>\n",
       "      <td>0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>99.043518</td>\n",
       "      <td>796.189221</td>\n",
       "      <td>601.950952</td>\n",
       "      <td>1224.586350</td>\n",
       "      <td>428.397129</td>\n",
       "      <td>502.907433</td>\n",
       "      <td>1.173928</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0.999927</td>\n",
       "      <td>108.871812</td>\n",
       "      <td>798.997039</td>\n",
       "      <td>594.052243</td>\n",
       "      <td>1221.967910</td>\n",
       "      <td>422.970871</td>\n",
       "      <td>485.180431</td>\n",
       "      <td>1.147078</td>\n",
       "      <td>205217.189787</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>0410e7f5-4bfd-4d35-b623-9f8d6b87c179/0003</td>\n",
       "      <td>0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>98.499827</td>\n",
       "      <td>792.350204</td>\n",
       "      <td>604.166475</td>\n",
       "      <td>1221.011094</td>\n",
       "      <td>428.660889</td>\n",
       "      <td>505.666648</td>\n",
       "      <td>1.179643</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0.999941</td>\n",
       "      <td>112.773276</td>\n",
       "      <td>792.317748</td>\n",
       "      <td>592.349200</td>\n",
       "      <td>1215.951043</td>\n",
       "      <td>423.633296</td>\n",
       "      <td>479.575925</td>\n",
       "      <td>1.132054</td>\n",
       "      <td>203164.329518</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>0410e7f5-4bfd-4d35-b623-9f8d6b87c179/0004</td>\n",
       "      <td>0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>99.756652</td>\n",
       "      <td>789.944644</td>\n",
       "      <td>605.358793</td>\n",
       "      <td>1219.450434</td>\n",
       "      <td>429.505790</td>\n",
       "      <td>505.602141</td>\n",
       "      <td>1.177172</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0.999928</td>\n",
       "      <td>111.382233</td>\n",
       "      <td>791.457046</td>\n",
       "      <td>593.168555</td>\n",
       "      <td>1211.594545</td>\n",
       "      <td>420.137499</td>\n",
       "      <td>481.786322</td>\n",
       "      <td>1.146735</td>\n",
       "      <td>202416.500180</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>5 rows × 31 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "                                  guid/image  cocov1-can1-label  \\\n",
       "0  0410e7f5-4bfd-4d35-b623-9f8d6b87c179/0000                  0   \n",
       "1  0410e7f5-4bfd-4d35-b623-9f8d6b87c179/0001                  0   \n",
       "2  0410e7f5-4bfd-4d35-b623-9f8d6b87c179/0002                  0   \n",
       "3  0410e7f5-4bfd-4d35-b623-9f8d6b87c179/0003                  0   \n",
       "4  0410e7f5-4bfd-4d35-b623-9f8d6b87c179/0004                  0   \n",
       "\n",
       "   cocov1-can1-score  cocov1-can1-ymin  cocov1-can1-xmin  cocov1-can1-ymax  \\\n",
       "0                1.0        105.870206        778.004116        599.913759   \n",
       "1                1.0        107.833644        768.235507        598.237929   \n",
       "2                1.0         99.043518        796.189221        601.950952   \n",
       "3                1.0         98.499827        792.350204        604.166475   \n",
       "4                1.0         99.756652        789.944644        605.358793   \n",
       "\n",
       "   cocov1-can1-xmax  cocov1-can1-box-width  cocov1-can1-box-height  \\\n",
       "0       1198.261631             420.257514              494.043552   \n",
       "1       1182.570957             414.335451              490.404285   \n",
       "2       1224.586350             428.397129              502.907433   \n",
       "3       1221.011094             428.660889              505.666648   \n",
       "4       1219.450434             429.505790              505.602141   \n",
       "\n",
       "   cocov1-can1-box-ratio         ...           incep-can1-label  \\\n",
       "0               1.175573         ...                          0   \n",
       "1               1.183592         ...                          0   \n",
       "2               1.173928         ...                          0   \n",
       "3               1.179643         ...                          0   \n",
       "4               1.177172         ...                          0   \n",
       "\n",
       "   incep-can1-score  incep-can1-ymin  incep-can1-xmin  incep-can1-ymax  \\\n",
       "0          0.999922       114.727974       778.541677       597.486985   \n",
       "1          0.999919       118.921772       768.581806       588.000484   \n",
       "2          0.999927       108.871812       798.997039       594.052243   \n",
       "3          0.999941       112.773276       792.317748       592.349200   \n",
       "4          0.999928       111.382233       791.457046       593.168555   \n",
       "\n",
       "   incep-can1-xmax  incep-can1-box-width  incep-can1-box-height  \\\n",
       "0      1199.429844            420.888167             482.759011   \n",
       "1      1186.015816            417.434010             469.078712   \n",
       "2      1221.967910            422.970871             485.180431   \n",
       "3      1215.951043            423.633296             479.575925   \n",
       "4      1211.594545            420.137499             481.786322   \n",
       "\n",
       "   incep-can1-box-ratio  incep-can1-box-area  \n",
       "0              1.147001        203187.555194  \n",
       "1              1.123719        195809.407618  \n",
       "2              1.147078        205217.189787  \n",
       "3              1.132054        203164.329518  \n",
       "4              1.146735        202416.500180  \n",
       "\n",
       "[5 rows x 31 columns]"
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "cocov1_kitti = cocov1.merge(kitti,how='left',on='guid/image',sort=True)\n",
    "cocov1_kitti_incep = cocov1_kitti.merge(incep,how='left',on='guid/image',sort=True)\n",
    "cocov1_kitti_incep.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "cocov1_kitti_incep.to_csv('lgb_test.tsv',sep='\\t',index=False)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 2",
   "language": "python",
   "name": "python2"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 2
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython2",
   "version": "2.7.14"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}