notebooks/01_get_started/tracking_with_bells_and_whistles.ipynb

{
 "cells": [
  {
   "attachments": {},
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Tracking with bells and whistles"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "%load_ext autoreload\n",
    "%autoreload 2"
   ]
  },
  {
   "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 tr th {\n",
       "        text-align: left;\n",
       "    }\n",
       "\n",
       "    .dataframe thead tr:last-of-type th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr>\n",
       "      <th>TeamID</th>\n",
       "      <th colspan=\"10\" halign=\"left\">0</th>\n",
       "      <th>...</th>\n",
       "      <th colspan=\"5\" halign=\"left\">1</th>\n",
       "      <th colspan=\"5\" halign=\"left\">BALL</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>PlayerID</th>\n",
       "      <th colspan=\"5\" halign=\"left\">0</th>\n",
       "      <th colspan=\"5\" halign=\"left\">1</th>\n",
       "      <th>...</th>\n",
       "      <th colspan=\"5\" halign=\"left\">9</th>\n",
       "      <th colspan=\"5\" halign=\"left\">BALL</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Attributes</th>\n",
       "      <th>bb_height</th>\n",
       "      <th>bb_left</th>\n",
       "      <th>bb_top</th>\n",
       "      <th>bb_width</th>\n",
       "      <th>conf</th>\n",
       "      <th>bb_height</th>\n",
       "      <th>bb_left</th>\n",
       "      <th>bb_top</th>\n",
       "      <th>bb_width</th>\n",
       "      <th>conf</th>\n",
       "      <th>...</th>\n",
       "      <th>bb_height</th>\n",
       "      <th>bb_left</th>\n",
       "      <th>bb_top</th>\n",
       "      <th>bb_width</th>\n",
       "      <th>conf</th>\n",
       "      <th>bb_height</th>\n",
       "      <th>bb_left</th>\n",
       "      <th>bb_top</th>\n",
       "      <th>bb_width</th>\n",
       "      <th>conf</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>frame</th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>39.0</td>\n",
       "      <td>801.00</td>\n",
       "      <td>1051.00</td>\n",
       "      <td>39.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>39.0</td>\n",
       "      <td>1464.0</td>\n",
       "      <td>1080.000000</td>\n",
       "      <td>39.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>...</td>\n",
       "      <td>39.0</td>\n",
       "      <td>2537.000000</td>\n",
       "      <td>1146.000000</td>\n",
       "      <td>39.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>10.00</td>\n",
       "      <td>1916.00</td>\n",
       "      <td>1073.00</td>\n",
       "      <td>10.00</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>39.0</td>\n",
       "      <td>801.75</td>\n",
       "      <td>1052.25</td>\n",
       "      <td>39.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>39.0</td>\n",
       "      <td>1464.0</td>\n",
       "      <td>1079.692308</td>\n",
       "      <td>39.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>...</td>\n",
       "      <td>39.0</td>\n",
       "      <td>2536.857143</td>\n",
       "      <td>1145.714286</td>\n",
       "      <td>39.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>9.75</td>\n",
       "      <td>1910.75</td>\n",
       "      <td>1073.75</td>\n",
       "      <td>9.75</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>39.0</td>\n",
       "      <td>802.50</td>\n",
       "      <td>1053.50</td>\n",
       "      <td>39.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>39.0</td>\n",
       "      <td>1464.0</td>\n",
       "      <td>1079.384615</td>\n",
       "      <td>39.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>...</td>\n",
       "      <td>39.0</td>\n",
       "      <td>2536.714286</td>\n",
       "      <td>1145.428571</td>\n",
       "      <td>39.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>9.50</td>\n",
       "      <td>1905.50</td>\n",
       "      <td>1074.50</td>\n",
       "      <td>9.50</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>39.0</td>\n",
       "      <td>803.25</td>\n",
       "      <td>1054.75</td>\n",
       "      <td>39.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>39.0</td>\n",
       "      <td>1464.0</td>\n",
       "      <td>1079.076923</td>\n",
       "      <td>39.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>...</td>\n",
       "      <td>39.0</td>\n",
       "      <td>2536.571429</td>\n",
       "      <td>1145.142857</td>\n",
       "      <td>39.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>9.25</td>\n",
       "      <td>1900.25</td>\n",
       "      <td>1075.25</td>\n",
       "      <td>9.25</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>39.0</td>\n",
       "      <td>804.00</td>\n",
       "      <td>1056.00</td>\n",
       "      <td>39.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>39.0</td>\n",
       "      <td>1464.0</td>\n",
       "      <td>1078.769231</td>\n",
       "      <td>39.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>...</td>\n",
       "      <td>39.0</td>\n",
       "      <td>2536.428571</td>\n",
       "      <td>1144.857143</td>\n",
       "      <td>39.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>9.00</td>\n",
       "      <td>1895.00</td>\n",
       "      <td>1076.00</td>\n",
       "      <td>9.00</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>5 rows × 115 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "TeamID             0                                                   \\\n",
       "PlayerID           0                                        1           \n",
       "Attributes bb_height bb_left   bb_top bb_width conf bb_height bb_left   \n",
       "frame                                                                   \n",
       "1               39.0  801.00  1051.00     39.0  1.0      39.0  1464.0   \n",
       "2               39.0  801.75  1052.25     39.0  1.0      39.0  1464.0   \n",
       "3               39.0  802.50  1053.50     39.0  1.0      39.0  1464.0   \n",
       "4               39.0  803.25  1054.75     39.0  1.0      39.0  1464.0   \n",
       "5               39.0  804.00  1056.00     39.0  1.0      39.0  1464.0   \n",
       "\n",
       "TeamID                                 ...         1               \\\n",
       "PlayerID                               ...         9                \n",
       "Attributes       bb_top bb_width conf  ... bb_height      bb_left   \n",
       "frame                                  ...                          \n",
       "1           1080.000000     39.0  1.0  ...      39.0  2537.000000   \n",
       "2           1079.692308     39.0  1.0  ...      39.0  2536.857143   \n",
       "3           1079.384615     39.0  1.0  ...      39.0  2536.714286   \n",
       "4           1079.076923     39.0  1.0  ...      39.0  2536.571429   \n",
       "5           1078.769231     39.0  1.0  ...      39.0  2536.428571   \n",
       "\n",
       "TeamID                                     BALL                             \\\n",
       "PlayerID                                   BALL                              \n",
       "Attributes       bb_top bb_width conf bb_height  bb_left   bb_top bb_width   \n",
       "frame                                                                        \n",
       "1           1146.000000     39.0  1.0     10.00  1916.00  1073.00    10.00   \n",
       "2           1145.714286     39.0  1.0      9.75  1910.75  1073.75     9.75   \n",
       "3           1145.428571     39.0  1.0      9.50  1905.50  1074.50     9.50   \n",
       "4           1145.142857     39.0  1.0      9.25  1900.25  1075.25     9.25   \n",
       "5           1144.857143     39.0  1.0      9.00  1895.00  1076.00     9.00   \n",
       "\n",
       "TeamID           \n",
       "PlayerID         \n",
       "Attributes conf  \n",
       "frame            \n",
       "1           1.0  \n",
       "2           1.0  \n",
       "3           1.0  \n",
       "4           1.0  \n",
       "5           1.0  \n",
       "\n",
       "[5 rows x 115 columns]"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import sportslabkit as slk\n",
    "from sportslabkit.logger import show_df # This just makes the df viewable in the notebook.\n",
    "\n",
    "\n",
    "dataset_path = slk.datasets.get_path(\"top_view\")\n",
    "path_to_csv = sorted(dataset_path.glob(\"annotations/*.csv\"))[0]\n",
    "path_to_mp4 = sorted(dataset_path.glob(\"videos/*.mp4\"))[0]\n",
    "\n",
    "bbdf = slk.load_df(path_to_csv)\n",
    "\n",
    "bbdf.head()"
   ]
  },
  {
   "attachments": {},
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "To use the full soccertrack dataset, see [\"Dataset Preparation\"](../02_user_guide/dataset_preparation.ipynb)."
   ]
  },
  {
   "attachments": {},
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Tracking setup"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "Tracking Progress: 100%|██████████| 10/10 [01:02<00:00,  6.25s/it, Active: 9, Dead: 0]\n"
     ]
    }
   ],
   "source": [
    "from sportslabkit.utils import get_git_root\n",
    "from sportslabkit.mot import TeamTracker\n",
    "from sportslabkit.mot.callbacks import TeamClassificationCallback\n",
    "\n",
    "root = get_git_root()\n",
    "cam = slk.Camera(path_to_mp4)\n",
    "\n",
    "det_model = slk.detection_model.load('YOLOv8x', model=root / 'models/yolov8/soccer_top_view_best.pt', imgsz=1280)\n",
    "motion_model = slk.motion_model.load('GCLinear')\n",
    "calibration_model = slk.calibration_model.load('LineBasedCalibrator', min_line_length=50, line_distance_threshold=50, line_thickness=15, morph_size=15)\n",
    "image_model = slk.image_model.load('resnet50')\n",
    "\n",
    "first_matching_fn = slk.matching.MotionVisualMatchingFunction(\n",
    "    motion_metric=slk.metrics.EuclideanCMM2D(use_pred_pt=True),\n",
    "    motion_metric_gate=0.9,\n",
    "    visual_metric=slk.metrics.CosineCMM(),\n",
    "    visual_metric_gate=0.5,\n",
    "    beta=0.9,\n",
    ")\n",
    "\n",
    "second_matching_fn = slk.matching.SimpleMatchingFunction(\n",
    "    metric=slk.metrics.EuclideanCMM2D(use_pred_pt=True),\n",
    "    gate=0.5,\n",
    ")\n",
    "\n",
    "vector_model = slk.vector_model.load('sklearnvectormodel', model_path='https://drive.google.com/file/d/1d6SskM1VFc_kIjFbQqEKgnMrmVnDRwWZ/view')\n",
    "callbacks = [TeamClassificationCallback(vector_model)]\n",
    "\n",
    "tracker = TeamTracker(\n",
    "    detection_model=det_model,\n",
    "    image_model=image_model,\n",
    "    motion_model=motion_model,\n",
    "    calibration_model=calibration_model,\n",
    "    first_matching_fn=first_matching_fn,\n",
    "    second_matching_fn=second_matching_fn,\n",
    "    detection_score_threshold=0.5,\n",
    "    max_staleness=3,\n",
    "    min_length=5,\n",
    "    callbacks=callbacks,\n",
    ")\n",
    "\n",
    "res = tracker.track(cam[:10])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "Writing video: 10it [00:00, 10.83it/s]\n"
     ]
    }
   ],
   "source": [
    "save_path = \"assets/tracking_results_with_teamtrack.mp4\"\n",
    "res.visualize_frames(cam.video_path, save_path)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "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 tr th {\n",
       "        text-align: left;\n",
       "    }\n",
       "\n",
       "    .dataframe thead tr:last-of-type th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr>\n",
       "      <th>TeamID</th>\n",
       "      <th colspan=\"10\" halign=\"left\">0</th>\n",
       "      <th>...</th>\n",
       "      <th colspan=\"8\" halign=\"left\">1</th>\n",
       "      <th colspan=\"2\" halign=\"left\">BALL</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>PlayerID</th>\n",
       "      <th colspan=\"2\" halign=\"left\">0</th>\n",
       "      <th colspan=\"2\" halign=\"left\">1</th>\n",
       "      <th colspan=\"2\" halign=\"left\">10</th>\n",
       "      <th colspan=\"2\" halign=\"left\">2</th>\n",
       "      <th colspan=\"2\" halign=\"left\">3</th>\n",
       "      <th>...</th>\n",
       "      <th colspan=\"2\" halign=\"left\">6</th>\n",
       "      <th colspan=\"2\" halign=\"left\">7</th>\n",
       "      <th colspan=\"2\" halign=\"left\">8</th>\n",
       "      <th colspan=\"2\" halign=\"left\">9</th>\n",
       "      <th colspan=\"2\" halign=\"left\">BALL</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Attributes</th>\n",
       "      <th>x</th>\n",
       "      <th>y</th>\n",
       "      <th>x</th>\n",
       "      <th>y</th>\n",
       "      <th>x</th>\n",
       "      <th>y</th>\n",
       "      <th>x</th>\n",
       "      <th>y</th>\n",
       "      <th>x</th>\n",
       "      <th>y</th>\n",
       "      <th>...</th>\n",
       "      <th>x</th>\n",
       "      <th>y</th>\n",
       "      <th>x</th>\n",
       "      <th>y</th>\n",
       "      <th>x</th>\n",
       "      <th>y</th>\n",
       "      <th>x</th>\n",
       "      <th>y</th>\n",
       "      <th>x</th>\n",
       "      <th>y</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>frame</th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>13.153587</td>\n",
       "      <td>34.164146</td>\n",
       "      <td>36.983471</td>\n",
       "      <td>35.303577</td>\n",
       "      <td>53.197151</td>\n",
       "      <td>34.666561</td>\n",
       "      <td>52.224819</td>\n",
       "      <td>15.419998</td>\n",
       "      <td>40.538166</td>\n",
       "      <td>46.167290</td>\n",
       "      <td>...</td>\n",
       "      <td>67.093147</td>\n",
       "      <td>15.537871</td>\n",
       "      <td>52.840797</td>\n",
       "      <td>43.730240</td>\n",
       "      <td>64.841843</td>\n",
       "      <td>45.449814</td>\n",
       "      <td>74.867973</td>\n",
       "      <td>37.785244</td>\n",
       "      <td>52.547379</td>\n",
       "      <td>34.098305</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>13.179957</td>\n",
       "      <td>34.208954</td>\n",
       "      <td>36.983601</td>\n",
       "      <td>35.292648</td>\n",
       "      <td>53.099606</td>\n",
       "      <td>34.701454</td>\n",
       "      <td>52.295338</td>\n",
       "      <td>15.411969</td>\n",
       "      <td>40.591766</td>\n",
       "      <td>46.167347</td>\n",
       "      <td>...</td>\n",
       "      <td>67.093216</td>\n",
       "      <td>15.523928</td>\n",
       "      <td>52.836475</td>\n",
       "      <td>43.716923</td>\n",
       "      <td>64.841843</td>\n",
       "      <td>45.448952</td>\n",
       "      <td>74.863007</td>\n",
       "      <td>37.775181</td>\n",
       "      <td>52.357082</td>\n",
       "      <td>34.115082</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>13.206327</td>\n",
       "      <td>34.253761</td>\n",
       "      <td>36.983730</td>\n",
       "      <td>35.281727</td>\n",
       "      <td>53.002056</td>\n",
       "      <td>34.736347</td>\n",
       "      <td>52.365852</td>\n",
       "      <td>15.403940</td>\n",
       "      <td>40.645363</td>\n",
       "      <td>46.167404</td>\n",
       "      <td>...</td>\n",
       "      <td>67.093277</td>\n",
       "      <td>15.509986</td>\n",
       "      <td>52.832150</td>\n",
       "      <td>43.703606</td>\n",
       "      <td>64.841850</td>\n",
       "      <td>45.448086</td>\n",
       "      <td>74.858047</td>\n",
       "      <td>37.765121</td>\n",
       "      <td>52.166763</td>\n",
       "      <td>34.131859</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>13.232697</td>\n",
       "      <td>34.298569</td>\n",
       "      <td>36.983864</td>\n",
       "      <td>35.270798</td>\n",
       "      <td>52.904499</td>\n",
       "      <td>34.771244</td>\n",
       "      <td>52.436363</td>\n",
       "      <td>15.395912</td>\n",
       "      <td>40.698959</td>\n",
       "      <td>46.167461</td>\n",
       "      <td>...</td>\n",
       "      <td>67.093346</td>\n",
       "      <td>15.496042</td>\n",
       "      <td>52.827827</td>\n",
       "      <td>43.690292</td>\n",
       "      <td>64.841850</td>\n",
       "      <td>45.447227</td>\n",
       "      <td>74.853088</td>\n",
       "      <td>37.755058</td>\n",
       "      <td>51.976418</td>\n",
       "      <td>34.148636</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>13.259068</td>\n",
       "      <td>34.343376</td>\n",
       "      <td>36.983994</td>\n",
       "      <td>35.259872</td>\n",
       "      <td>52.806931</td>\n",
       "      <td>34.806145</td>\n",
       "      <td>52.506870</td>\n",
       "      <td>15.387884</td>\n",
       "      <td>40.752552</td>\n",
       "      <td>46.167519</td>\n",
       "      <td>...</td>\n",
       "      <td>67.093407</td>\n",
       "      <td>15.482101</td>\n",
       "      <td>52.823502</td>\n",
       "      <td>43.676975</td>\n",
       "      <td>64.841858</td>\n",
       "      <td>45.446362</td>\n",
       "      <td>74.848114</td>\n",
       "      <td>37.744999</td>\n",
       "      <td>51.786053</td>\n",
       "      <td>34.165417</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>5 rows × 46 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "TeamID              0                                                         \\\n",
       "PlayerID            0                     1                    10              \n",
       "Attributes          x          y          x          y          x          y   \n",
       "frame                                                                          \n",
       "1           13.153587  34.164146  36.983471  35.303577  53.197151  34.666561   \n",
       "2           13.179957  34.208954  36.983601  35.292648  53.099606  34.701454   \n",
       "3           13.206327  34.253761  36.983730  35.281727  53.002056  34.736347   \n",
       "4           13.232697  34.298569  36.983864  35.270798  52.904499  34.771244   \n",
       "5           13.259068  34.343376  36.983994  35.259872  52.806931  34.806145   \n",
       "\n",
       "TeamID                                                  ...          1  \\\n",
       "PlayerID            2                     3             ...          6   \n",
       "Attributes          x          y          x          y  ...          x   \n",
       "frame                                                   ...              \n",
       "1           52.224819  15.419998  40.538166  46.167290  ...  67.093147   \n",
       "2           52.295338  15.411969  40.591766  46.167347  ...  67.093216   \n",
       "3           52.365852  15.403940  40.645363  46.167404  ...  67.093277   \n",
       "4           52.436363  15.395912  40.698959  46.167461  ...  67.093346   \n",
       "5           52.506870  15.387884  40.752552  46.167519  ...  67.093407   \n",
       "\n",
       "TeamID                                                                        \\\n",
       "PlayerID                       7                     8                     9   \n",
       "Attributes          y          x          y          x          y          x   \n",
       "frame                                                                          \n",
       "1           15.537871  52.840797  43.730240  64.841843  45.449814  74.867973   \n",
       "2           15.523928  52.836475  43.716923  64.841843  45.448952  74.863007   \n",
       "3           15.509986  52.832150  43.703606  64.841850  45.448086  74.858047   \n",
       "4           15.496042  52.827827  43.690292  64.841850  45.447227  74.853088   \n",
       "5           15.482101  52.823502  43.676975  64.841858  45.446362  74.848114   \n",
       "\n",
       "TeamID                      BALL             \n",
       "PlayerID                    BALL             \n",
       "Attributes          y          x          y  \n",
       "frame                                        \n",
       "1           37.785244  52.547379  34.098305  \n",
       "2           37.775181  52.357082  34.115082  \n",
       "3           37.765121  52.166763  34.131859  \n",
       "4           37.755058  51.976418  34.148636  \n",
       "5           37.744999  51.786053  34.165417  \n",
       "\n",
       "[5 rows x 46 columns]"
      ]
     },
     "execution_count": 17,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "codf = bbdf.to_codf(tracker.homographies[0], method='bottom_middle')\n",
    "codf.head()"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "soccertrack-KNSgM8DD-py3.9",
   "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.10.0"
  },
  "orig_nbformat": 4,
  "vscode": {
   "interpreter": {
    "hash": "977a6a4298a6f74f911a0aa33bb1949bba79863f90698d4cbd5dc1167a3d3b34"
   }
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}