Merge pull request #2716 from invigorzz313/foregroundExtract

foreground extraction from images

Author: avinashkranjan

Foreground Extraction/ReadMe.md

@@ -0,0 +1,20 @@
+# Foreground extraction
+This python script will allow us to extract foreground from an image using Grabcut algorithm.
+
+## Setup Instructions
+### Install python3
+sudo apt-get install python3
+### Install pip (package installer for python)
+sudo apt-get install python3-pip
+### Install Numpy library with pip
+pip3 install numpy
+### Install OpenCV library with pip
+pip3 install opencv-python
+### Install tkinter library
+sudo apt-get install python3-tk
+
+## Details/Output
+User is expected to select an image and mark a rectangle around the foreground that they wish to extract. The script then runs and the extracted foreground using Grabcut algorithm is then displayed and saved.  
+
+## Author
+Github: invigorzz313

Foreground Extraction/Sample_Image.jpg

@@ -0,0 +1,53 @@
+import cv2
+import numpy as np
+import tkinter as tk
+from tkinter.filedialog import *
+
+# asking input from user
+photo = askopenfilename()
+img = cv2.imread(photo)
+imgcopy = img
+
+a,b = -1,-1
+rect = (0,0,0,0)
+draw = False
+count = 0
+
+# for selecting the required foreground portion with a rectangle
+def rect_draw(event, x, y, flag, param):
+    global draw,a,b,rect,count
+    if (event == cv2.EVENT_LBUTTONDOWN):
+        draw = True
+        a,b = x,y
+    elif (event == cv2.EVENT_MOUSEMOVE):
+        if (draw==True):
+            rect = (a,b,x,y)
+    elif (event == cv2.EVENT_LBUTTONUP):
+        draw = False
+        count = 1
+        cv2.rectangle(img, (a,b),(x,y), (0,255,0), 1)
+        rect = (a,b,x,y)
+        cv2.imshow('Mark foreground', img)
+        cv2.waitKey(0)
+        cv2.destroyAllWindows()
+
+cv2.namedWindow('Mark foreground')
+cv2.setMouseCallback('Mark foreground', rect_draw) # call back function
+while count==0:
+    cv2.imshow('Mark foreground', img)
+    if cv2.waitKey(20) & 0xFF == 27:
+        break
+
+# Using the grabcut algorithm
+mask = np.zeros(imgcopy.shape[:2], np.uint8)
+bgdModel = np.zeros((1,65), np.float64)
+fgdModel = np.zeros((1,65), np.float64)
+cv2.grabCut(imgcopy, mask, rect, bgdModel, fgdModel, 5, cv2.GC_INIT_WITH_RECT)
+mask2 = np.where((mask==2)|(mask==0),0,1).astype('uint8')
+imgcopy = imgcopy*mask2[:,:,np.newaxis]
+
+# displaying and saving the output image
+cv2.imshow("Output", imgcopy)
+cv2.imwrite("Output.png", imgcopy)
+cv2.waitKey(0)
+cv2.destroyAllWindows()