test_6_objects.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "import numpy as np\n",
    "import cv2\n",
    "\n",
    "green = (0,255,0)\n",
    "red = (0,0,255)\n",
    "white = (255,255,255)\n",
    "fonts = cv2.FONT_HERSHEY_COMPLEX\n",
    "cap = cv2.VideoCapture(0)\n",
    "\n",
    "#########################################################################\n",
    "\n",
    "width = 960\n",
    "height= 720\n",
    "cap.set(4,height)\n",
    "cap.set(3,width)\n",
    "\n",
    "#########################################################################\n",
    "\n",
    "def empty(a):\n",
    "    pass\n",
    "\n",
    "#########################################################################\n",
    "\n",
    "cv2.namedWindow(\"Parameters\")\n",
    "cv2.resizeWindow(\"Parameters\",1280,480)\n",
    "cv2.createTrackbar(\"Threshold1\",\"Parameters\",70,255,empty)\n",
    "cv2.createTrackbar(\"Threshold2\",\"Parameters\",70,255,empty)\n",
    "cv2.createTrackbar(\"Area\",\"Parameters\",1000,10000,empty)\n",
    "\n",
    "#########################################################################\n",
    "\n",
    "def getContours(img,imgContour):\n",
    "    \n",
    "# find contours function\n",
    "# retrieval method\n",
    "# chanprox: all contour points\n",
    "    contours, hierarchy = cv2.findContours(img, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)\n",
    "    \n",
    "    for cnt in contours:\n",
    "        area = cv2.contourArea(cnt)\n",
    "        areaMin = cv2.getTrackbarPos(\"Area\", \"Parameters\")        \n",
    "        if area > areaMin:\n",
    "#             cv2.drawContours(imgContour,cnt,-1,red,7)\n",
    "            peri = cv2.arcLength(cnt, True)\n",
    "            approx = cv2.approxPolyDP(cnt, 0.02 * peri, True)\n",
    "#             print(len(approx))\n",
    "            x,y,w,h = cv2.boundingRect(approx)\n",
    "    \n",
    "            cv2.rectangle(imgContour,(x,y),(x+w,y+h),green,3)\n",
    "            distance = -0.03*(area/100) + 37\n",
    "\n",
    "#             # Display rectangle\n",
    "#             box = cv2.boxPoints(rect)\n",
    "#             box = np.int0(box)\n",
    "\n",
    "#             cv2.circle(img,(int(x),int(y)),5,red,-1)\n",
    "#             cv2.polylines(img,[box],True,(255,0,0),2)\n",
    "\n",
    "#             cv2.putText(imgContour,\"Points: \" + str(len(approx)),(x+w+20,y+20),fonts,0.7,green,2)\n",
    "#             cv2.putText(imgContour,\"Height: \" + str(int(h)),(x+w+20,y),fonts,0.7,green,2)\n",
    "#             cv2.putText(imgContour,\"Width: \" + str(int(w)),(x+w+20,y+25),fonts,0.7,green,2)\n",
    "#             cv2.putText(imgContour,\"Aspect: \" + str(round((h/w),1)),(x+w+20,y+25),fonts,0.7,green,2)\n",
    "#             cv2.putText(imgContour,\"Distance: \" + str(int(distance)),(x+w+20,y+25),fonts,0.7,green,2)\n",
    "            cv2.putText(imgContour,\"Area: \" + str(int(area)),(x+w+20,y+50),fonts,0.7,green,2)\n",
    "        \n",
    "            if(area < 8300):\n",
    "                cv2.putText(imgContour,\"Grasp: Pinch\",(x+w+20,y+75),fonts,0.7,green,2)\n",
    "            elif(area > 15000):\n",
    "                cv2.putText(imgContour,\"Grasp: Power\",(x+w+20,y+75),fonts,0.7,green,2)\n",
    "            else:\n",
    "                cv2.putText(imgContour,\"Grasp: Tripod\",(x+w+20,y+75),fonts,0.7,green,2)\n",
    "    \n",
    "#########################################################################    \n",
    "\n",
    "while(True):        \n",
    "    success, img = cap.read()    # Capture frame-by-frame\n",
    "    imgContour = img\n",
    "    \n",
    "# imgBlur = cv2.GaussianBlur(img,(7,7),1)\n",
    "    imgGray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)\n",
    "    \n",
    "    \n",
    "# Canny edge detector is an edge detection operator that uses a multi-stage algorithm\n",
    "# to detect a wide range of edges in images\n",
    "\n",
    "# canny image takes 2 thresholds (upper and lower)\n",
    "# If a pixel gradient is higher than the upper threshold, it is accepted as an edge.\n",
    "# If it is between the two thresholds, it will be accepted only if it is connected\n",
    "# to a pixel that is above the upper threshold.\n",
    "\n",
    "# get trackbar position from the parameters window\n",
    "    threshold1 = cv2.getTrackbarPos(\"Threshold1\",\"Parameters\")\n",
    "    threshold2 = cv2.getTrackbarPos(\"Threshold2\",\"Parameters\")\n",
    "    imgCanny = cv2.Canny(imgGray,threshold1,threshold2)\n",
    "    \n",
    "# dilation function to remove overlaps and noise    \n",
    "    kernel = np.ones((5,5))\n",
    "    imgDil = cv2.dilate(imgCanny, kernel, iterations=1)\n",
    "    \n",
    "# contour: outline representing or bounding the shape    \n",
    "    getContours(imgDil, imgContour)\n",
    "    \n",
    "#     imgStack = stackImages(0.8,([img, imgBlur, imgGray], [imgDil,imgDil,imgDil]))\n",
    "\n",
    "    cv2.imshow('frame',imgContour)\n",
    "    \n",
    "#########################################################################    \n",
    "\n",
    "    if cv2.waitKey(1)==27: break    # esc\n",
    "\n",
    "cap.release()\n",
    "cv2.destroyAllWindows()"
   ]
  }
 ],
 "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.8.5"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}