working fine for trajectory with crash at end

Author: kmr0877

object_track.ipynb

@@ -31,33 +31,45 @@
  },
  "outputs": [],
  "source": [
- "def process_image(im):\n",
- " image, contours, hier = cv2.findContours(im, cv2.RETR_TREE,\n",
- " cv2.CHAIN_APPROX_SIMPLE)\n",
- "\n",
- " # with each contour, draw boundingRect in green\n",
- " # a minAreaRect in red and\n",
- " # a minEnclosingCircle in blue\n",
- " for c in contours:\n",
+ "greenLower = (29, 86, 6)\n",
+ "greenUpper = (64, 255, 255)\n",
+ " \n",
+ "# initialize the list of tracked points, the frame counter,\n",
+ "# and the coordinate deltas\n",
+ "pts = deque(maxlen=20)\n",
+ "counter = 0\n",
+ "(dX, dY) = (0, 0)\n",
+ "direction = \"\"\n",
+ "filename = 'test_sample2.avi'\n",
+ "vid = imageio.get_reader(filename, 'ffmpeg')\n",
+ "count = 0\n",
+ "for num in range(vid.get_length()):\n",
+ " try:\n",
+ " img = vid.get_data(num)\n",
  " \n",
- " # get the bounding rect\n",
- " x, y, w, h = cv2.boundingRect(c)\n",
+ " img = cv2.resize(img,(1000,600))\n",
+ " gray_image = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)\n",
+ " _,im = cv2.threshold(gray_image,220,1,cv2.THRESH_BINARY+cv2.THRESH_OTSU)\n",
  " \n",
- " # draw a green rectangle to visualize the bounding rect\n",
- " cv2.rectangle(img, (x, y), (x+w, y+h), (0, 255, 0), 2)\n",
+ " image, cnts, hier = cv2.findContours(im,cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)\n",
+ " center = None\n",
+ " for c in cnts:\n",
+ " x, y, w, h = cv2.boundingRect(c)\n",
  "\n",
- " # get the min area rect\n",
- " rect = cv2.minAreaRect(c)\n",
- " box = cv2.boxPoints(rect)\n",
- " \n",
- " # convert all coordinates floating point values to int\n",
- " box = np.int0(box)\n",
- " \n",
- " # draw a red 'nghien' rectangle\n",
- " cv2.drawContours(img, [box], 0, (0, 0, 255))\n",
+ "  # draw a green rectangle to visualize the bounding rect\n",
+ "  cv2.rectangle(img, (x, y), (x+w, y+h), (0, 255, 0), 2)\n",
+ "\n",
+ "  # get the min area rect\n",
+ "  rect = cv2.minAreaRect(c)\n",
+ "  box = cv2.boxPoints(rect)\n",
+ "\n",
+ "  # convert all coordinates floating point values to int\n",
+ "  box = np.int0(box)\n",
  "\n",
- " cv2.drawContours(img, contours, -1, (255, 255, 0), 1)\n",
- " return img\n"
+ " # draw a red 'nghien' rectangle\n",
+ " cv2.drawContours(img, [box], 0, (0, 0, 255))\n",
+ " # only proceed if at least one contour was found\n",
+ "\n"
  ]
  },
  {