video_tracking.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "from src.application.utils import generate_bounding_boxes, split_video_to_frames, merge_frames_to_video, calculate_histogram_bounding_box, calculate_histogram_distance, calculate_histogram_from_coordinates, check_bounding_boxes\n",
    "\n",
    "import os"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "data_path = os.path.join('src', 'application', 'data')\n",
    "video_path = os.path.join(data_path, 'video_hamilton.mp4')\n",
    "output_frames_path = os.path.join(data_path, 'out_frames')\n",
    "bboxed_frames_path = os.path.join(output_frames_path, 'bounding_box')\n",
    "model_ckpt_path = os.path.join(data_path, 'model.ckpt')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "split_video_to_frames(video_path, output_frames_path, 4)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "outputs = generate_bounding_boxes(model_ckpt_path, output_frames_path)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "Reading frame_0372.jpg: 100%|██████████| 373/373 [00:04<00:00, 91.60it/s]\n",
      "Writing frame #372: 100%|██████████| 373/373 [00:03<00:00, 119.98it/s]\n"
     ]
    }
   ],
   "source": [
    "merge_frames_to_video(bboxed_frames_path, data_path, fps=30)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "import json, math\n",
    "\"\"\"\n",
    "images = os.listdir(bboxed_frames_path)\n",
    "for i, image_name in enumerate(images):\n",
    "    if i == len(os.listdir(bboxed_frames_path)) -1:\n",
    "        break\n",
    "    \n",
    "    histogram_frame_curr = calculate_histogram_bounding_box(os.path.join(bboxed_frames_path, image_name), os.path.join(output_frames_path, 'txt_bounding_box', image_name.replace('.jpg', '.txt')))\n",
    "    \n",
    "    histogram_frame_next = calculate_histogram_bounding_box(os.path.join(bboxed_frames_path, images[i+1]), os.path.join(output_frames_path, 'txt_bounding_box', images[i+1].replace('.jpg', '.txt')))\n",
    "    \n",
    "    print(calculate_histogram_distance(histogram_frame_curr[1], histogram_frame_next[1]))\n",
    "    \n",
    "    break\n",
    "\"\"\"\n",
    "\n",
    "images = ['frame_0073.jpg', 'frame_0074.jpg']\n",
    "bounding_boxes = {0: [], 1: []}\n",
    "centroids = {0: [], 1: []}\n",
    "for i, image_name in enumerate(images):\n",
    "    with open(os.path.join(output_frames_path, 'txt_bounding_box', image_name.replace('.jpg', '.txt')), mode='r') as f:\n",
    "        bounding_boxes[i] = json.load(f)\n",
    "    \n",
    "    for bbox in bounding_boxes[i]:\n",
    "        centroids[i].append(((bbox[0] + bbox[2]) / 2, bbox[1] + bbox[3] / 2))\n",
    "        \n",
    "checking = True\n",
    "while checking:\n",
    "    \n",
    "    if len(centroids[0]) == len(centroids[1]):\n",
    "        continue\n",
    "    \n",
    "    for idx, c1 in enumerate(centroids[0]):\n",
    "        for c2 in centroids[1]:\n",
    "            distance = math.hypot(c2[0] - c1[0], c2[1] - c2[1])\n",
    "            if distance < 30:\n",
    "                continue\n",
    "            \n",
    "            histogram_frame_curr = calculate_histogram_from_coordinates(os.path.join(bboxed_frames_path, images[0]), bounding_boxes[0][idx])\n",
    "            histogram_frame_next = calculate_histogram_from_coordinates(os.path.join(bboxed_frames_path, images[1]), bounding_boxes[0][idx])\n",
    "            \n",
    "            hist_distance = calculate_histogram_distance(histogram_frame_curr[0], histogram_frame_next[0])\n",
    "            \n",
    "            if hist_distance < 5:\n",
    "                pass # draw the bounding box.\n",
    "            \n",
    "             \n",
    "    checking = False\n",
    "    \n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "4.0533555685440295\n",
      "[[972.0225830078125, 186.95899963378906, 1042.819091796875, 226.9969940185547]] [[971.1002807617188, 190.26077270507812, 1041.67529296875, 231.5342254638672]]\n"
     ]
    }
   ],
   "source": [
    "check_bounding_boxes(output_frames_path)"
   ]
  }
 ],
 "metadata": {
  "interpreter": {
   "hash": "07efdcd4b820c98a756949507a4d29d7862823915ec7477944641bea022f4f62"
  },
  "kernelspec": {
   "display_name": "Python 3.9.7 64-bit ('base': conda)",
   "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.9.7"
  },
  "orig_nbformat": 4
 },
 "nbformat": 4,
 "nbformat_minor": 2
}