third_party/opencv
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Minor update in grabcut.py

Browse Source
This commit is contained in:
Muhammad Taha 2020-02-18 10:22:41 +05:00 committed by Alexander Smorkalov
parent 30331eef4b
commit 8177adbd54
1 changed files with 7 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
samples/python/grabcut.py
View File

@ -11,10 +11,10 @@ USAGE:
README FIRST: README FIRST:
Two windows will show up, one for input and one for output. Two windows will show up, one for input and one for output.
At first, in input window, draw a rectangle around the object using At first, in input window, draw a rectangle around the object using the
mouse right button. Then press 'n' to segment the object (once or a few times) right mouse button. Then press 'n' to segment the object (once or a few times)
For any finer touch-ups, you can press any of the keys below and draw lines on For any finer touch-ups, you can press any of the keys below and draw lines on
the areas you want. Then again press 'n' for updating the output. the areas you want. Then again press 'n' to update the output.
Key '0' - To select areas of sure background Key '0' - To select areas of sure background
Key '1' - To select areas of sure foreground Key '1' - To select areas of sure foreground
@ -44,8 +44,8 @@ class App():
DRAW_BG = {'color' : BLACK, 'val' : 0} DRAW_BG = {'color' : BLACK, 'val' : 0}
DRAW_FG = {'color' : WHITE, 'val' : 1} DRAW_FG = {'color' : WHITE, 'val' : 1}
DRAW_PR_FG = {'color' : GREEN, 'val' : 3}
DRAW_PR_BG = {'color' : RED, 'val' : 2} DRAW_PR_BG = {'color' : RED, 'val' : 2}
DRAW_PR_FG = {'color' : GREEN, 'val' : 3}
# setting up flags # setting up flags
rect = (0,0,1,1) rect = (0,0,1,1)
@ -160,14 +160,12 @@ class App():
print(""" For finer touchups, mark foreground and background after pressing keys 0-3 print(""" For finer touchups, mark foreground and background after pressing keys 0-3
and again press 'n' \n""") and again press 'n' \n""")
try: try:
bgdmodel = np.zeros((1, 65), np.float64)
fgdmodel = np.zeros((1, 65), np.float64)
if (self.rect_or_mask == 0): # grabcut with rect if (self.rect_or_mask == 0): # grabcut with rect
bgdmodel = np.zeros((1, 65), np.float64)
fgdmodel = np.zeros((1, 65), np.float64)
cv.grabCut(self.img2, self.mask, self.rect, bgdmodel, fgdmodel, 1, cv.GC_INIT_WITH_RECT) cv.grabCut(self.img2, self.mask, self.rect, bgdmodel, fgdmodel, 1, cv.GC_INIT_WITH_RECT)
self.rect_or_mask = 1 self.rect_or_mask = 1
elif self.rect_or_mask == 1: # grabcut with mask elif (self.rect_or_mask == 1): # grabcut with mask
bgdmodel = np.zeros((1, 65), np.float64)
fgdmodel = np.zeros((1, 65), np.float64)
cv.grabCut(self.img2, self.mask, self.rect, bgdmodel, fgdmodel, 1, cv.GC_INIT_WITH_MASK) cv.grabCut(self.img2, self.mask, self.rect, bgdmodel, fgdmodel, 1, cv.GC_INIT_WITH_MASK)
except: except:
import traceback import traceback