iterating over tracked points done for direction

Author: kmr0877

object_track.ipynb

@@ -91,7 +91,46 @@
  " cv2.circle(img, (int(x), int(y)), int(radius),\n",
  " (0, 255, 255), 2)\n",
  " cv2.circle(img, center, 5, (0, 0, 255), -1)\n",
- " pts.appendleft(center)"
+ " pts.appendleft(center)\n",
+ " # loop over the set of tracked points\n",
+ " for i in np.arange(1, len(pts)):\n",
+ " # if either of the tracked points are None, ignore\n",
+ " # them\n",
+ " if pts[i - 1] is None or pts[i] is None:\n",
+ " continue\n",
+ " # check to see if enough points have been accumulated in\n",
+ " # the buffer\n",
+ " if counter >= 10 and i == 1 and pts[-10] is not None:\n",
+ " # compute the difference between the x and y\n",
+ " # coordinates and re-initialize the direction\n",
+ " # text variables\n",
+ " dX = pts[-10][0] - pts[i][0]\n",
+ " dY = pts[-10][1] - pts[i][1]\n",
+ " (dirX, dirY) = (\"\", \"\")\n",
+ "\n",
+ " # ensure there is significant movement in the\n",
+ " # x-direction\n",
+ " if np.abs(dX) > 20:\n",
+ " dirX = \"East\" if np.sign(dX) == 1 else \"West\"\n",
+ "\n",
+ " # ensure there is significant movement in the\n",
+ " # y-direction\n",
+ " if np.abs(dY) > 20:\n",
+ " dirY = \"North\" if np.sign(dY) == 1 else \"South\"\n",
+ "\n",
+ " # handle when both directions are non-empty\n",
+ " if dirX != \"\" and dirY != \"\":\n",
+ " direction = \"{}-{}\".format(dirY, dirX)\n",
+ "\n",
+ " # otherwise, only one direction is non-empty\n",
+ " else:\n",
+ " direction = dirX if dirX != \"\" else dirY\n",
+ " # otherwise, compute the thickness of the line and\n",
+ " # draw the connecting lines\n",
+ " thickness = int(np.sqrt(15 / float(i + 1)) * 2.5)\n",
+ " cv2.line(img, pts[i - 1], pts[i], (0, 0, 255), thickness)\n",
+ "\n",
+ " "
  ]
  },
  {
@@ -243,7 +282,7 @@
  "name": "python",
  "nbconvert_exporter": "python",
  "pygments_lexer": "ipython3",
- "version": "3.6.1"
+ "version": "3.5.4"
  }
  },
  "nbformat": 4,