object_tracking added

Author: ajithvcoder

python/object_tracking/Object Tracking.ipynb

@@ -0,0 +1,115 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Object Tracking"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import cv2\n",
+    "import numpy as np\n",
+    "\n",
+    "# camera is inirialised\n",
+    "capture = cv2.VideoCapture(0)\n",
+    "\n",
+    "# range of HSV color space\n",
+    "lower_purple=np.array([130,50,90])\n",
+    "upper_purple = np.array([170,255,255])\n",
+    "\n",
+    "# to store the center points\n",
+    "points=[]\n",
+    "\n",
+    "# get frame size\n",
+    "ret,frame=capture.read()\n",
+    "Height,Width = frame.shape[:2]\n",
+    "frame_count=0\n",
+    "\n",
+    "while True:\n",
+    "    # cam frame is read\n",
+    "    ret,frame = capture.read()\n",
+    "    hsv_img = cv2.cvtColor(frame,cv2.COLOR_BGR2HSV)\n",
+    "\n",
+    "    # threshodl to get only purple colors\n",
+    "    mask =cv2.inRange(hsv_img,lower_purple,upper_purple)\n",
+    "\n",
+    "    # find countors\n",
+    "    contours,_ = cv2.findContours(mask.copy(),cv2.RETR_EXTERNAL,\n",
+    "                                  cv2.CHAIN_APPROX_SIMPLE)\n",
+    "    # get the centroid of mass\n",
+    "    center = int(Height/2),int(Width/2)\n",
+    "\n",
+    "    if len(contours)>0:\n",
+    "        # get only the lasrgest contour\n",
+    "        c=max(contours,key=cv2.contourArea)\n",
+    "        (x,y ),radius = cv2.minEnclosingCircle(c)\n",
+    "        M=cv2.moments(c)\n",
+    "\n",
+    "        try:\n",
+    "            center=(int(M[\"m10\"]/M[\"m00\"]),int(M[\"m01\"]/M[\"m00\"]))\n",
+    "        except:\n",
+    "            center=int(Height/2),int(Width/2)\n",
+    "\n",
+    "        # draw circle only if its greater than a particular size\n",
+    "        if radius>25:\n",
+    "            cv2.circle(frame,(int(x),int(y)),int(radius),(0,0,255))\n",
+    "            cv2.circle(frame,center,5,(0,255,0),-1)\n",
+    "\n",
+    "\n",
+    "    points.append(center)\n",
+    "\n",
+    "    # loop over the points\n",
+    "    if radius>25:\n",
+    "        for i in range(1,len(points)):\n",
+    "            try:\n",
+    "                cv2.line(frame,points[i-1],points[i],(0,255,0),2)\n",
+    "            except:\n",
+    "                pass\n",
+    "        frame_count = 0\n",
+    "\n",
+    "    else:\n",
+    "        # count the frames\n",
+    "        frame_count+=1\n",
+    "        # if its 10 frames then trails are deleted\n",
+    "        if frame_count==10:\n",
+    "            points=[]\n",
+    "            frame_count=0\n",
+    "\n",
+    "    frame = cv2.flip(frame,1)\n",
+    "    cv2.imshow(\"Object tracker\",frame)\n",
+    "\n",
+    "    if cv2.waitKey(1)==13:\n",
+    "        break\n",
+    "capture.release()\n",
+    "cv2.destroyAllWindows()\n"
+   ]
+  }
+ ],
+ "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
+}