measurments_tracking_and_timing.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "import matplotlib.pyplot as plt\n",
    "import numpy as np\n",
    "import cv2\n",
    "import datetime\n",
    "from datetime import time,datetime\n",
    "\n",
    "def gammaCorrection(frame, gamma):\n",
    "    invGamma = 1 / gamma\n",
    "    table = [((i / 255) ** invGamma) * 255 for i in range(256)]\n",
    "    table = np.array(table, np.uint8)\n",
    "    return cv2.LUT(frame, table)\n",
    "\n",
    "def get_binary_image(frame):\n",
    "    enhanced_frame = gammaCorrection(frame, 2)\n",
    "    gray_frame = cv2.GaussianBlur(cv2.cvtColor(enhanced_frame, cv2.COLOR_RGB2GRAY),\n",
    "                                  (5,5),0)\n",
    "    threshold = cv2.threshold(gray_frame, 200,255, cv2.THRESH_BINARY)[1]\n",
    "    return (enhanced_frame, threshold)\n",
    "\n",
    "def display_tracking_window_axis_center(frame,x,y,tw_width,tw_height, rows):\n",
    "        bounding_box = cv2.rectangle(frame,\n",
    "                  (x, y),\n",
    "                  (x + tw_width, y + tw_height),\n",
    "                  (255, 0, 0), 2)\n",
    "        horisontal_axis = cv2.line(frame,\n",
    "                        (x + int(tw_width/2), 0),\n",
    "                        (x + int(tw_width/2), rows),\n",
    "                        (255, 0, 0), 2)\n",
    "        object_center = cv2.circle(frame,\n",
    "                                (x + int(tw_width/2), y + int(tw_height/2)),\n",
    "                                2, (255,0,0),2)\n",
    "        \n",
    "        return (bounding_box, horisontal_axis,object_center)\n",
    "\n",
    "def draw_lines(frame, l, y, tw_height):\n",
    "     line = [((l-1), y), ((l-1), tw_height),((l+1), tw_height),((l+1), y)]\n",
    "     cv2.polylines(frame,[np.array(line, np.int32)],True, (0,0,255), 1)\n",
    "\n",
    "\n",
    "def check_line_crossing(l,x,y,tw_width,tw_height):\n",
    "    line = [((l-1), y), ((l-1), tw_height),((l+1), tw_height),((l+1), y)]\n",
    "    line_crossing = cv2.pointPolygonTest(\n",
    "        np.array(line, np.int32),\n",
    "        ((x + int(tw_width/2),\n",
    "          y + int(tw_height/2))),\n",
    "        False)\n",
    "    \n",
    "    return line_crossing"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "video = cv2.VideoCapture(r'.\\input_data\\2023_03_14_PTM.avi')\n",
    "\n",
    "fullsize_first_frame = video.read()[1]\n",
    "\n",
    "roi_start_x = 315\n",
    "roi_end_x = 370\n",
    "roi_start_y = 207\n",
    "roi_end_y = 400\n",
    "first_frame = fullsize_first_frame[roi_start_x:roi_end_x, roi_start_y:roi_end_y]\n",
    "\n",
    "%matplotlib widget\n",
    "plt.imshow(fullsize_first_frame)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "video = cv2.VideoCapture(r'.\\input_data\\2023_03_14_ptm.avi')\n",
    "\n",
    "roi_start_x = 315\n",
    "roi_end_x = 370\n",
    "roi_start_y = 207\n",
    "roi_end_y = 400\n",
    "\n",
    "tw_start_point_x = 30\n",
    "tw_start_point_y = 0\n",
    "tw_start_width = 25\n",
    "tw_start_height = 60\n",
    "term_criteria= (cv2.TERM_CRITERIA_EPS | cv2.TERM_CRITERIA_COUNT, 10,1)\n",
    "\n",
    "lines_list = [9,25,41,55,71,87,102,119,133,149,164,180]\n",
    "line_labels = ['-6','-5','-4','-3','-2','-1','0','1','2','3','4','5']\n",
    "\n",
    "while True:\n",
    "    orig_frame = video.read()[1]\n",
    "    frame = orig_frame[roi_start_x:roi_end_x, roi_start_y:roi_end_y]  \n",
    "    enhanced_frame, threshold = get_binary_image(frame)\n",
    "    track_window = cv2.meanShift(threshold,\n",
    "                                 (tw_start_point_x, tw_start_point_y,\n",
    "                                  tw_start_width, tw_start_height),\n",
    "                                 term_criteria)[1]\n",
    "    \n",
    "    rows, cols, _bands = enhanced_frame.shape\n",
    "    tw_start_point_x, tw_start_point_y, tw_width, tw_height = track_window\n",
    "    tracking_window_geometry = display_tracking_window_axis_center(\n",
    "         enhanced_frame,\n",
    "         tw_start_point_x, tw_start_point_y,\n",
    "         tw_width, tw_height,rows)\n",
    "    bounding_box, horisontal_axis, object_center = tracking_window_geometry\n",
    "\n",
    "    line_crossing_check = []\n",
    "    for lines in lines_list:\n",
    "        draw_lines(enhanced_frame,lines,tw_start_point_y,tw_height)\n",
    "        crossing = check_line_crossing(lines,\n",
    "                                       tw_start_point_x, tw_start_point_y,\n",
    "                                       tw_width, tw_height)\n",
    "        line_crossing_check.append(crossing)\n",
    "\n",
    "    for line, next_line, scale_label in zip(line_crossing_check[1:11],\n",
    "                                            line_crossing_check[2:11],\n",
    "                                            line_labels[2:11]):\n",
    "            if line == 1:\n",
    "                initial_scale_time = datetime.now()\n",
    "                initial_scale_timestamp = datetime.timestamp(initial_scale_time)\n",
    "            if next_line == 1:\n",
    "                next_scale_time = datetime.now()\n",
    "                next_scale_timestamp = datetime.timestamp(next_scale_time)\n",
    "                scale_crossing = next_scale_timestamp-initial_scale_timestamp\n",
    "                scale_crossing = round(scale_crossing,3)\n",
    "                \n",
    "                crossing_index = line_crossing_check.index(next_line)\n",
    "                crossing_index = scale_label\n",
    "                print(f'Scale:{scale_label} Seconds:{scale_crossing}')                \n",
    "\n",
    "    cv2.imshow('Frame', enhanced_frame)\n",
    "    key = cv2.waitKey(100) #for framerate 20, normal playback is 100\n",
    "    if key == 27:\n",
    "        break\n",
    "video.release()\n",
    "cv2.destroyAllWindows()"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "base",
   "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.1"
  },
  "orig_nbformat": 4
 },
 "nbformat": 4,
 "nbformat_minor": 2
}